Keluar

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

Lapisan putus sekolah.

Dropout terdiri dari pengaturan acak sebagian kecil unit input ke 0 pada setiap pembaruan selama waktu pelatihan, yang membantu mencegah overfitting.

  • Pernyataan

    public typealias TangentVector = EmptyTangentVector
  • Pernyataan

    @noDerivative
    public let probability: Double
  • Membuat lapisan putus sekolah.

    Prasyarat

    probabilitas harus bernilai antara 0 dan 1 (inklusif).

    Pernyataan

    public init(probability: Double)

    Parameter

    probability

    Kemungkinan sebuah node putus.

  • 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.