Trung bình3D

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

Lớp tổng hợp trung bình cho dữ liệu không gian hoặc không gian-thời gian.

  • Tuyên ngôn

    public typealias TangentVector = EmptyTangentVector
  • Kích thước của cửa sổ giảm trượt để gộp.

    Tuyên ngôn

    @noDerivative
    public let poolSize: (Int, Int, Int, Int, Int)
  • Các bước của cửa sổ trượt cho từng chiều của đầu vào 5-D. Các bước tiến trong các chiều không gian phải là 1 .

    Tuyên ngôn

    @noDerivative
    public let strides: (Int, Int, Int, Int, Int)
  • Thuật toán đệm để gộp chung.

    Tuyên ngôn

    @noDerivative
    public let padding: Padding
  • Tạo một lớp tổng hợp trung bình.

    Tuyên ngôn

    public init(
      poolSize: (Int, Int, Int, Int, Int),
      strides: (Int, Int, Int, Int, Int),
      padding: Padding
    )
  • Trả về kết quả thu được từ việc áp dụng lớp cho đầu vào đã cho.

    Tuyên ngôn

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

    Thông số

    input

    Đầu vào của lớp.

    Giá trị trả về

    Đầu ra.

  • Tạo một lớp tổng hợp trung bình.

    Tuyên ngôn

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

    Thông số

    poolSize

    Các yếu tố theo chiều dọc và chiều ngang để giảm quy mô.

    strides

    Những bước tiến.

    padding

    Phần đệm.

  • Tạo một lớp gộp trung bình với kích thước và bước tiến của cửa sổ gộp được chỉ định. Tất cả các kích thước và bước tiến gộp đều giống nhau.

    Tuyên ngôn

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