1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
| #pragma once
|
| #include <torch/arg.h>
| #include <torch/csrc/WindowsTorchApiMacro.h>
| #include <torch/types.h>
|
| namespace torch {
| namespace nn {
|
| /// Options for the `Linear` module.
| struct TORCH_API LinearOptions {
| LinearOptions(int64_t in, int64_t out);
| /// The number of input features (columns of the input matrix).
| TORCH_ARG(int64_t, in);
| /// The number of output features to produce (columns of the output matrix).
| TORCH_ARG(int64_t, out);
| /// Whether to learn and add a bias after the linear transformation.
| TORCH_ARG(bool, with_bias) = true;
| };
|
| } // namespace nn
| } // namespace torch
|
|
