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

Commit e900edb

Browse files
committed
Adding doc for Filter shapes
1 parent dfcc60d commit e900edb

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -658,8 +658,10 @@ public struct SeparableConv2D<Scalar: TensorFlowFloatingPoint>: Layer {
658658
/// bias, activation function, strides, and padding.
659659
///
660660
/// - Parameters:
661-
/// - depthwiseFilter: The 4-D depthwise convolution kernel.
662-
/// - pointwiseFilter: The 4-D pointwise convolution kernel.
661+
/// - depthwiseFilter: The 4-D depthwise convolution kernel
662+
/// `[filter height, filter width, input channels count, channel multiplier]`.
663+
/// - pointwiseFilter: The 4-D pointwise convolution kernel
664+
/// `[1, 1, channel multiplier * input channels count, output channels count]`.
663665
/// - bias: The bias vector.
664666
/// - activation: The element-wise activation function.
665667
/// - strides: The strides of the sliding window for spatial dimensions.

0 commit comments

Comments
 (0)