1 #include <torch/nn/options/linear.h>
2
3 namespace torch {
4 namespace nn {
5
LinearOptions(int64_t in_features,int64_t out_features)6 LinearOptions::LinearOptions(int64_t in_features, int64_t out_features)
7 : in_features_(in_features), out_features_(out_features) {}
8
BilinearOptions(int64_t in1_features,int64_t in2_features,int64_t out_features)9 BilinearOptions::BilinearOptions(
10 int64_t in1_features,
11 int64_t in2_features,
12 int64_t out_features)
13 : in1_features_(in1_features),
14 in2_features_(in2_features),
15 out_features_(out_features) {}
16
UnflattenOptions(int64_t dim,std::vector<int64_t> sizes)17 UnflattenOptions::UnflattenOptions(int64_t dim, std::vector<int64_t> sizes)
18 : dim_(dim), sizes_(std::move(sizes)) {}
19
UnflattenOptions(const char * dimname,namedshape_t namedshape)20 UnflattenOptions::UnflattenOptions(const char* dimname, namedshape_t namedshape)
21 : dim_(0),
22 dimname_(std::string(dimname)),
23 namedshape_(std::move(namedshape)) {}
24
UnflattenOptions(std::string dimname,namedshape_t namedshape)25 UnflattenOptions::UnflattenOptions(std::string dimname, namedshape_t namedshape)
26 : dim_(0),
27 dimname_(std::move(dimname)),
28 namedshape_(std::move(namedshape)) {}
29
30 } // namespace nn
31 } // namespace torch
32