Options for Dropout
and FeatureDropout
.
More...
#include <dropout.h>
Public Member Functions | |
DropoutOptions (double rate=0.5) | |
TORCH_ARG (double, rate) | |
The probability with which a particular component of the input is set to zero. More... | |
torch::nn::DropoutOptions::TORCH_ARG | ( | double | , |
rate | |||
) |
The probability with which a particular component of the input is set to zero.
Changes to this parameter at runtime are effective.