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.
1.8.11