Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit a969648

Browse files
committed
Updating to refactored conv3dbackpropinput'
1 parent f1f1da1 commit a969648

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -545,10 +545,9 @@ public struct TransposedConv3D: Layer {
545545
strides.2 + (filter.shape[2] * paddingIndex)
546546
let c = filter.shape[3]
547547
let newShape = Tensor<Int32>([Int32(batchSize), Int32(w), Int32(h), Int32(d), Int32(c)])
548-
return activation(input.conv2DBackpropInput(shape: newShape, filter: filter,
549-
strides: (1, strides.0, strides.1,
550-
strides.2, 1),
551-
padding: padding) + bias)
548+
return activation(conv3DBackpropInput(input, shape: newShape, filter: filter,
549+
strides: (1, strides.0, strides.1,strides.2, 1),
550+
padding: padding) + bias)
552551
}
553552
}
554553

0 commit comments

Comments
 (0)