Abandonar

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

Una capa de abandono.

Deserción consiste en establecer aleatoriamente una fracción de las unidades de entrada a 0 en cada actualización durante el tiempo de formación, lo que ayuda a evitar el sobreajuste.

  • Declaración

    public typealias TangentVector = EmptyTangentVector

  • Declaración

    @noDerivative
public let probability: Double

  • Crea una capa de abandono.

    Condición previa

    la probabilidad debe ser un valor entre 0 y 1 (inclusive).

    Declaración

    public init(probability: Double)

    Parámetros

    probability

    La probabilidad de que un nodo se caiga.

  • Devuelve la salida obtenida al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor devuelto

    La salida.