Trung bình2D 

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

Một lớp tổng hợp trung bình cho dữ liệu không 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)

  • Các bước của cửa sổ trượt cho từng chiều của đầu vào 4-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)

  • 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), strides: (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), strides: (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.