평균Pool2D

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

공간 데이터에 대한 평균 풀링 계층입니다.

  • 선언

    public typealias TangentVector = EmptyTangentVector
  • 풀링을 위한 슬라이딩 감소 창의 크기입니다.

    선언

    @noDerivative
    public let poolSize: (Int, Int, Int, Int)
  • 4차원 입력의 각 차원에 대한 슬라이딩 윈도우의 보폭입니다. 비공간 차원의 스트라이드는 1 이어야 합니다.

    선언

    @noDerivative
    public let strides: (Int, Int, Int, Int)
  • 풀링을 위한 패딩 알고리즘입니다.

    선언

    @noDerivative
    public let padding: Padding
  • 평균 풀링 계층을 생성합니다.

    선언

    public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding)
  • 주어진 입력에 레이어를 적용하여 얻은 출력을 반환합니다.

    선언

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

    매개변수

    input

    레이어에 대한 입력입니다.

    반환 값

    출력.

  • 평균 풀링 계층을 생성합니다.

    선언

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

    매개변수

    poolSize

    축소할 수직 및 수평 요소입니다.

    strides

    진전.

    padding

    패딩.