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

Commit ccf7fcc

Browse files
jon-towrxwei
authored andcommitted
Update parameter documentation for TransposedConv2D (#283)
1 parent aeb2874 commit ccf7fcc

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -417,8 +417,9 @@ public struct TransposedConv2D: Layer {
417417
/// activation function, strides, and padding.
418418
///
419419
/// - Parameters:
420-
/// - filter: The 4-D convolution kernel.
421-
/// - bias: The bias vector.
420+
/// - filter: A 4-D tensor of shape
421+
/// `[width, height, input channel count, output channel count]`.
422+
/// - bias: The bias tensor of shape `[output channel count]`.
422423
/// - activation: The element-wise activation function.
423424
/// - strides: The strides of the sliding window for spatial dimensions.
424425
/// - padding: The padding algorithm for convolution.
@@ -465,7 +466,8 @@ public extension TransposedConv2D {
465466
/// initialization with the specified generator. The bias vector is initialized with zeros.
466467
///
467468
/// - Parameters:
468-
/// - filterShape: The shape of the 4-D convolution kernel.
469+
/// - filterShape: A 4-D tensor of shape
470+
/// `[width, height, input channel count, output channel count]`.
469471
/// - strides: The strides of the sliding window for spatial dimensions.
470472
/// - padding: The padding algorithm for convolution.
471473
/// - activation: The element-wise activation function.
@@ -497,7 +499,8 @@ public extension TransposedConv2D {
497499
/// initialization with the specified seed. The bias vector is initialized with zeros.
498500
///
499501
/// - Parameters:
500-
/// - filterShape: The shape of the 4-D convolution kernel.
502+
/// - filterShape: A 4-D tensor of shape
503+
/// `[width, height, input channel count, output channel count]`.
501504
/// - strides: The strides of the sliding window for spatial dimensions.
502505
/// - padding: The padding algorithm for convolution.
503506
/// - activation: The element-wise activation function.

0 commit comments

Comments
 (0)