PoolMéd.2D

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

Uma camada média de pooling para dados espaciais.

  • Declaração

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

    Declaração

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

    Declaração

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

    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 ao aplicar a 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 a escala.

    strides

    Os avanços.

    padding

    O preenchimento.