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

Commit ef35d21

Browse files
sjaz24rxwei
authored andcommitted
Fix 'TransposedConv2D' initializer. (#334)
See https://bugs.swift.org/browse/TF-540 for background on this issue.
1 parent 09c68d7 commit ef35d21

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -443,7 +443,7 @@ public extension TransposedConv2D {
443443
filterShape.0, filterShape.1, filterShape.2, filterShape.3])
444444
self.init(
445445
filter: filterInitializer(filterTensorShape),
446-
bias: biasInitializer([filterShape.3]),
446+
bias: biasInitializer([filterShape.2]),
447447
activation: activation,
448448
strides: strides,
449449
padding: padding)

0 commit comments

Comments
 (0)