俊逸星月

文章
6
资源
0
加入时间
3年0月8天

pytorch:nn.BatchNorm1d()用法介绍

torch.nn.BatchNorm1d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)num_features – 特征维度eps – 为数值稳定性而加到分母上的值。momentum – 移动平均默认的动量值。affine – 一个布尔值,当设置为真时,此模块具有可学习的仿射参数。这里其他几个参数都不重要,只需要看num_features就可以了。num_features就是你需要