Cair fora

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

Uma camada de eliminação.

O dropout consiste em definir aleatoriamente uma fração das unidades de entrada como 0 a cada atualização durante o tempo de treinamento, o que ajuda a evitar overfitting.

  • Declaração

    public typealias TangentVector = EmptyTangentVector
  • Declaração

    @noDerivative
    public let probability: Double
  • Cria uma camada suspensa.

    Condição prévia

    a probabilidade deve ser um valor entre 0 e 1 (inclusive).

    Declaração

    public init(probability: Double)

    Parâmetros

    probability

    A probabilidade de um nó cair.

  • Retorna a saída obtida ao aplicar a camada à entrada fornecida.

    Declaração

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

    Parâmetros

    input

    A entrada para a camada.

    Valor de retorno

    A saída.