Typedef torch::nn::BatchNorm2dOptions¶
Defined in File batchnorm.h
Typedef Documentation¶
-
using torch::nn::BatchNorm2dOptions = BatchNormOptions¶
Options for the
BatchNorm2d
module.Example:
BatchNorm2d model(BatchNorm2dOptions(4).eps(0.5).momentum(0.1).affine(false).track_running_stats(true));