탈락

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

드롭아웃 레이어.

드롭아웃은 훈련 시간 동안 업데이트될 때마다 입력 단위의 일부를 0 으로 무작위로 설정하는 것으로 구성되며, 이는 과적합을 방지하는 데 도움이 됩니다.

  • 선언

    public typealias TangentVector = EmptyTangentVector
  • 선언

    @noDerivative
    public let probability: Double
  • 드롭아웃 레이어를 생성합니다.

    전제조건

    확률은 0에서 1(포함) 사이의 값이어야 합니다.

    선언

    public init(probability: Double)

    매개변수

    probability

    노드가 삭제될 확률입니다.

  • 주어진 입력에 레이어를 적용하여 얻은 출력을 반환합니다.

    선언

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

    매개변수

    input

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

    반환 값

    출력.