Rata-rata Kolam2D

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

Lapisan pengumpulan rata-rata untuk data spasial.

  • Pernyataan

    public typealias TangentVector = EmptyTangentVector
  • Ukuran jendela reduksi geser untuk penggabungan.

    Pernyataan

    @noDerivative
    public let poolSize: (Int, Int, Int, Int)
  • Langkah jendela geser untuk setiap dimensi masukan 4-D. Langkah dalam dimensi non-spasial harus 1 .

    Pernyataan

    @noDerivative
    public let strides: (Int, Int, Int, Int)
  • Algoritma padding untuk pengumpulan.

    Pernyataan

    @noDerivative
    public let padding: Padding
  • Membuat lapisan pengumpulan rata-rata.

    Pernyataan

    public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)
  • Mengembalikan keluaran yang diperoleh dari penerapan lapisan ke masukan yang diberikan.

    Pernyataan

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

    Parameter

    input

    Masukan ke lapisan.

    Nilai Pengembalian

    Outputnya.

  • Membuat lapisan pengumpulan rata-rata.

    Pernyataan

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

    Parameter

    poolSize

    Faktor vertikal dan horizontal yang digunakan untuk menurunkan skala.

    strides

    Langkahnya.

    padding

    Bantalannya.