MédiaPool3D

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

Uma camada de pooling média para dados espaciais ou espaço-temporais.

  • 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, Int)
  • Os avanços da janela deslizante para cada dimensão de uma entrada 5-D. Os avanços em dimensões não espaciais devem ser 1 .

    Declaração

    @noDerivative
    public let strides: (Int, 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, Int),
      strides: (Int, 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, Int), strides: (Int, Int, Int), padding: Padding = .valid)

    Parâmetros

    poolSize

    Fatores verticais e horizontais pelos quais reduzir a escala.

    strides

    Os avanços.

    padding

    O preenchimento.

  • Cria uma camada de pooling média com o tamanho e a amplitude da janela de pooling especificados. Todos os tamanhos e avanços do pool são iguais.

    Declaração

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