pytorch：nn.BatchNorm1d()用法介绍

69 阅读 0 评论 46 点赞

我是靠谱客的博主俊逸星月，最近开发中收集的这篇文章主要介绍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就是你需要归一化的那一维的维度。
nn.BatchNorm1d本身不是给定输入矩阵，输出归一化结果的函数，而是定义了一个方法，再用这个方法去做归一化。
下面是一个例子。

BN = nn.BatchNorm1d(100)
input = torch.randn(20, 100)
output = m(input)

我们首先定义了一个归一化的函数BN，需要归一化的维度为100，其他参数为默认。然后随机初始化一个20×100的矩阵input，再用BN对这个矩阵归一化。
函数的input可以是二维或者三维。当input的维度为（N, C）时，BN将对C维归一化；当input的维度为(N, C, L) 时，归一化的维度同样为C维。