@frozen
public struct AvgPool2D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint
Una capa de agrupación promedio para datos espaciales.
Declaración
public typealias TangentVector = EmptyTangentVector
El tamaño de la ventana de reducción deslizante para pooling.
Declaración
@noDerivative public let poolSize: (Int, Int, Int, Int)
Los pasos de la ventana deslizante para cada dimensión de una entrada 4-D. Los pasos en dimensiones no espaciales deben ser
1
.Declaración
@noDerivative public let strides: (Int, Int, Int, Int)
El algoritmo de relleno para la agrupación.
Declaración
@noDerivative public let padding: Padding
Crea una capa de agrupación promedio.
Declaración
public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)
Crea una capa de agrupación promedio.
Declaración
public init(poolSize: (Int, Int), strides: (Int, Int), padding: Padding = .valid)
Parámetros
poolSize
Factores verticales y horizontales por los que reducir la escala.
strides
Los pasos.
padding
El acolchado.