Options for the Linear module.
More...
#include <linear.h>
Public Member Functions | |
| LinearOptions (int64_t in, int64_t out) | |
| TORCH_ARG (int64_t, in) | |
| The number of input features (columns of the input matrix). | |
| TORCH_ARG (int64_t, out) | |
| The number of output features to produce (columns of the output matrix). | |
| TORCH_ARG (bool, with_bias) | |
| Whether to learn and add a bias after the linear transformation. | |
1.8.11