xref: /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/batchnorm.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/nn/options/batchnorm.h>
2 
3 namespace torch {
4 namespace nn {
5 
BatchNormOptions(int64_t num_features)6 BatchNormOptions::BatchNormOptions(int64_t num_features)
7     : num_features_(num_features) {}
8 
9 } // namespace nn
10 } // namespace torch
11