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