@frozen
public struct AvgPool3D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint
Una capa de agrupación promedio para datos espaciales o espacio-temporales.
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, Int)
Los pasos de la ventana deslizante para cada dimensión de una entrada 5-D. Los pasos en dimensiones no espaciales deben ser
1
.Declaración
@noDerivative public let strides: (Int, 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, Int), strides: (Int, Int, Int, Int, Int), padding: Padding )
Crea una capa de agrupación promedio.
Declaración
public init(poolSize: (Int, Int, Int), strides: (Int, Int, Int), padding: Padding = .valid)
Parámetros
poolSize
Factores verticales y horizontales por los que reducir la escala.
strides
Los pasos.
padding
El acolchado.
Crea una capa de agrupación promedio con el tamaño y la zancada de la ventana de agrupación especificados. Todos los tamaños de agrupaciones y zancadas son iguales.
Declaración
public init(poolSize: Int, strides: Int, padding: Padding = .valid)