Abandonner

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

Une couche d'abandon.

L'abandon consiste à mettre aléatoirement à 0 une fraction d'unités d'entrée à chaque mise à jour pendant le temps d'entraînement, ce qui permet d'éviter le surapprentissage.

  • Déclaration

    public typealias TangentVector = EmptyTangentVector
  • Déclaration

    @noDerivative
    public let probability: Double
  • Crée un calque de suppression.

    Condition préalable

    la probabilité doit être une valeur comprise entre 0 et 1 (inclus).

    Déclaration

    public init(probability: Double)

    Paramètres

    probability

    La probabilité qu'un nœud abandonne.

  • Renvoie le résultat obtenu en appliquant le calque à l’entrée donnée.

    Déclaration

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

    Paramètres

    input

    L'entrée dans le calque.

    Valeur de retour

    Le résultat.