AvgPool2D

@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)
  • Devuelve el resultado obtenido al aplicar la capa a la entrada dada.

    Declaración

    @differentiable
    public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    Parámetros

    input

    La entrada a la capa.

    Valor de retorno

    La salida.

  • 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.