@frozen
public struct Dropout<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPointA dropout layer.
Dropout consists in randomly setting a fraction of input units to 0 at each update during
training time, which helps prevent overfitting.
-
Declaration
@noDerivative public let probability: Double -
Creates a dropout layer.
Precondition
probability must be a value between 0 and 1 (inclusive).Declaration
public init(probability: Double)Parameters
probabilityThe probability of a node dropping out.