AvgPool2D

@frozen
public struct AvgPool2D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint

Uma camada de agrupamento média para dados espaciais.

  • Declaração

    public typealias TangentVector = EmptyTangentVector
  • O tamanho da janela deslizante de redução para agrupamento.

    Declaração

    @noDerivative
    public let poolSize: (Int, Int, Int, Int)
  • As passadas da janela deslizante para cada dimensão de uma entrada 4-D. Avanços em dimensões espaciais não deve ser 1 .

    Declaração

    @noDerivative
    public let strides: (Int, Int, Int, Int)
  • O algoritmo de preenchimento para pool.

    Declaração

    @noDerivative
    public let padding: Padding
  • Cria uma camada de pooling média.

    Declaração

    public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)
  • Retorna a saída obtida da aplicação da camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.

  • Cria uma camada de pooling média.

    Declaração

    public init(poolSize: (Int, Int), strides: (Int, Int), padding: Padding = .valid)

    Parâmetros

    poolSize

    Fatores verticais e horizontais pelos quais reduzir.

    strides

    Os passos largos.

    padding

    O enchimento.