\((6)\) 其实BatchNorm1d,BatchNorm2d和BatchNorm3d的用法都是一样的,参数都只用传一个(即特征维度,手动实现的BatchNorm的num_features);对于手动实现的BatchNorm的初始化参数num_dims,是由1d,2d和3d来区分的 LazyBatchNorm系列连num_features都不用传,直接自动推断
BatchNorm1d
BatchNorm2d
BatchNorm3d
BatchNorm
num_features
num_dims
1d
2d
3d
LazyBatchNorm