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

Commit b2e3699

Browse files
Shashi456rxwei
authored andcommitted
Fix documentation errors in convolutional layers (#281)
1 parent 56d5c1b commit b2e3699

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@ public struct Conv1D<Scalar: TensorFlowFloatingPoint>: Layer {
3232
@noDerivative public let padding: Padding
3333
/// The dilation factor for the temporal dimension.
3434
@noDerivative public let dilation: Int
35-
36-
/// Creates a `Conv1D` layer with the specified filter, bias, activation function, stride, and
37-
/// padding.
35+
36+
/// Creates a `Conv1D` layer with the specified filter, bias, activation function, stride,
37+
/// dilation and padding.
3838
///
3939
/// - Parameters:
4040
/// - filter: The 3-D convolution kernel `[width, inputChannels, outputChannels]`.
@@ -76,7 +76,7 @@ public struct Conv1D<Scalar: TensorFlowFloatingPoint>: Layer {
7676
}
7777

7878
public extension Conv1D where Scalar.RawSignificand: FixedWidthInteger {
79-
/// Creates a `Conv1D` layer with the specified filter shape, stride, padding, and
79+
/// Creates a `Conv1D` layer with the specified filter shape, stride, padding, dilation and
8080
/// element-wise activation function. The filter tensor is initialized using Glorot uniform
8181
/// initialization with the specified generator. The bias vector is initialized with zeros.
8282
///
@@ -112,7 +112,7 @@ public extension Conv1D where Scalar.RawSignificand: FixedWidthInteger {
112112
}
113113

114114
public extension Conv1D {
115-
/// Creates a `Conv1D` layer with the specified filter shape, strides, padding, and
115+
/// Creates a `Conv1D` layer with the specified filter shape, strides, padding, dilation and
116116
/// element-wise activation function. The filter tensor is initialized using Glorot uniform
117117
/// initialization with the specified seed. The bias vector is initialized with zeros.
118118
///
@@ -163,19 +163,19 @@ public struct Conv2D<Scalar: TensorFlowFloatingPoint>: Layer {
163163
@noDerivative public let strides: (Int, Int)
164164
/// The padding algorithm for convolution.
165165
@noDerivative public let padding: Padding
166-
/// The dilation factor for spatials dimensions.
166+
/// The dilation factor for spatial dimensions.
167167
@noDerivative public let dilations: (Int, Int)
168-
169-
/// Creates a `Conv2D` layer with the specified filter, bias, activation function, strides, and
170-
/// padding.
168+
169+
/// Creates a `Conv2D` layer with the specified filter, bias, activation function, strides,
170+
/// dilations and padding.
171171
///
172172
/// - Parameters:
173173
/// - filter: The 4-D convolution kernel.
174174
/// - bias: The bias vector.
175175
/// - activation: The element-wise activation function.
176176
/// - strides: The strides of the sliding window for spatial dimensions.
177177
/// - padding: The padding algorithm for convolution.
178-
/// - dilations: The dilation factor for spatials dimensions.
178+
/// - dilations: The dilation factor for spatial dimensions.
179179
public init(
180180
filter: Tensor<Scalar>,
181181
bias: Tensor<Scalar>,
@@ -208,7 +208,7 @@ public struct Conv2D<Scalar: TensorFlowFloatingPoint>: Layer {
208208
}
209209

210210
public extension Conv2D {
211-
/// Creates a `Conv2D` layer with the specified filter shape, strides, padding, and
211+
/// Creates a `Conv2D` layer with the specified filter shape, strides, padding, dilations and
212212
/// element-wise activation function. The filter tensor is initialized using Glorot uniform
213213
/// initialization with the specified generator. The bias vector is initialized with zeros.
214214
///
@@ -243,7 +243,7 @@ public extension Conv2D {
243243
}
244244

245245
public extension Conv2D {
246-
/// Creates a `Conv2D` layer with the specified filter shape, strides, padding, and
246+
/// Creates a `Conv2D` layer with the specified filter shape, strides, padding, dilations and
247247
/// element-wise activation function. The filter tensor is initialized using Glorot uniform
248248
/// initialization with the specified seed. The bias vector is initialized with zeros.
249249
///

0 commit comments

Comments
 (0)