AvgPool2D

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

Un livello di pooling medio per i dati spaziali.

  • Dichiarazione

    public typealias TangentVector = EmptyTangentVector
  • La dimensione della finestra di riduzione scorrevole per il pooling.

    Dichiarazione

    @noDerivative
    public let poolSize: (Int, Int, Int, Int)
  • I passi della finestra scorrevole per ogni dimensione di un input 4-D. I passi nelle dimensioni non spaziali devono essere 1 .

    Dichiarazione

    @noDerivative
    public let strides: (Int, Int, Int, Int)
  • L'algoritmo di riempimento per il pooling.

    Dichiarazione

    @noDerivative
    public let padding: Padding
  • Crea un livello di pooling medio.

    Dichiarazione

    public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)
  • Restituisce l'output ottenuto dall'applicazione del livello all'input specificato.

    Dichiarazione

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

    Parametri

    input

    L'input al livello.

    Valore restituito

    L'uscita.

  • Crea un livello di pooling medio.

    Dichiarazione

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

    Parametri

    poolSize

    Fattori verticali e orizzontali attraverso i quali ridimensionare.

    strides

    I passi da gigante.

    padding

    L'imbottitura.