1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| #pragma once
|
| #include <torch/arg.h>
| #include <torch/csrc/WindowsTorchApiMacro.h>
| #include <torch/types.h>
|
| namespace torch {
| namespace nn {
|
| /// Options for `Dropout` and `FeatureDropout`.
| struct TORCH_API DropoutOptions {
| /* implicit */ DropoutOptions(double rate = 0.5);
| /// The probability with which a particular component of the input is set to
| /// zero.
| /// Changes to this parameter at runtime are effective.
| TORCH_ARG(double, rate);
| };
|
| } // namespace nn
| } // namespace torch
|
|
