Ritirarsi

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

Uno strato di abbandono.

Il dropout consiste nell'impostare casualmente una frazione di unità di input su 0 ad ogni aggiornamento durante il periodo di addestramento, il che aiuta a prevenire l'overfitting.

  • Dichiarazione

    public typealias TangentVector = EmptyTangentVector

  • Dichiarazione

    @noDerivative
public let probability: Double

  • Crea un livello di esclusione.

    Precondizione

    la probabilità deve essere un valore compreso tra 0 e 1 (incluso).

    Dichiarazione

    public init(probability: Double)

    Parametri

    probability

    La probabilità che un nodo cada.

  • Restituisce l'output ottenuto dall'applicazione del livello all'input specificato.

    Dichiarazione

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

    Parametri

    input

    L'input al livello.

    Valore di ritorno

    Il risultato.