@frozen
public struct AvgPool1D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint
An average pooling layer for temporal data.
-
Declaration
public typealias TangentVector = EmptyTangentVector
-
The size of the sliding reduction window for pooling.
Declaration
@noDerivative public let poolSize: Int
-
The stride of the sliding window for temporal dimension.
Declaration
@noDerivative public let stride: Int
-
The padding algorithm for pooling.
Declaration
@noDerivative public let padding: Padding
-
Creates an average pooling layer.
Declaration
public init(poolSize: Int, stride: Int, padding: Padding)
Parameters
poolSize
The size of the sliding reduction window for pooling.
stride
The stride of the sliding window for temporal dimension.
padding
The padding algorithm for pooling.