@@ -417,8 +417,9 @@ public struct TransposedConv2D: Layer {
417
417
/// activation function, strides, and padding.
418
418
///
419
419
/// - 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]`.
422
423
/// - activation: The element-wise activation function.
423
424
/// - strides: The strides of the sliding window for spatial dimensions.
424
425
/// - padding: The padding algorithm for convolution.
@@ -465,7 +466,8 @@ public extension TransposedConv2D {
465
466
/// initialization with the specified generator. The bias vector is initialized with zeros.
466
467
///
467
468
/// - 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]`.
469
471
/// - strides: The strides of the sliding window for spatial dimensions.
470
472
/// - padding: The padding algorithm for convolution.
471
473
/// - activation: The element-wise activation function.
@@ -497,7 +499,8 @@ public extension TransposedConv2D {
497
499
/// initialization with the specified seed. The bias vector is initialized with zeros.
498
500
///
499
501
/// - 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]`.
501
504
/// - strides: The strides of the sliding window for spatial dimensions.
502
505
/// - padding: The padding algorithm for convolution.
503
506
/// - activation: The element-wise activation function.
0 commit comments