Abandonar

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

Una capa de abandono.

El abandono consiste en establecer aleatoriamente una fracción de unidades de entrada en 0 en cada actualización durante el tiempo de entrenamiento, 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 el resultado obtenido 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 de retorno

    La salida.