public class Dropout
extends java.lang.Object
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.api.ndarray.INDArray |
applyDropConnect(Layer layer,
java.lang.String variable)
Apply drop connect to the given variable
|
static void |
applyDropout(org.nd4j.linalg.api.ndarray.INDArray input,
double dropout)
Apply dropout to the given input
and return the drop out mask used
|
public static org.nd4j.linalg.api.ndarray.INDArray applyDropConnect(Layer layer, java.lang.String variable)
layer
- the layer with the variablesvariable
- the variable to applypublic static void applyDropout(org.nd4j.linalg.api.ndarray.INDArray input, double dropout)
input
- the input to do drop out ondropout
- the drop out probability