xref: /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/adaptive.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/nn/options/adaptive.h>
2 
3 namespace torch {
4 namespace nn {
5 
AdaptiveLogSoftmaxWithLossOptions(int64_t in_features,int64_t n_classes,std::vector<int64_t> cutoffs)6 AdaptiveLogSoftmaxWithLossOptions::AdaptiveLogSoftmaxWithLossOptions(
7     int64_t in_features,
8     int64_t n_classes,
9     std::vector<int64_t> cutoffs)
10     : in_features_(in_features),
11       n_classes_(n_classes),
12       cutoffs_(std::move(cutoffs)) {}
13 
14 } // namespace nn
15 } // namespace torch
16