关于nn.BatchNorm1d()用法及说明
更新时间：2023年08月03日 16:27:35 作者：Code_demon
这篇文章主要介绍了关于nn.BatchNorm1d()用法及说明,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
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维。
使用nn.BatchNorm1d出现RuntimeError
pytorch模型中添加batchnorm过程中出现错误
RuntimeError: running_mean should contain 1 elements not 512
错误来自于nn.BatchNorm1d(N)，其中N=channels
解决办法
将输入重新排列
video = video.view(-1, video.shape[-1])
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家分享一个Python自动化办公的小工具——PDF拆分工具，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起动手试一试2022-03-03
PyCharm Anaconda配置PyQt5开发环境及创建项目的教程详解
这篇文章主要介绍了PyCharm Anaconda配置PyQt5开发环境及创建项目的方法，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-03-03
这篇文章主要介绍了pandas之query方法和sample随机抽样操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍了python中cv2.projectPoints的用法,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2023-12-12
这篇文章主要介绍了关于Pandas的Series创建方式和常用属性,Series 数据结构是用于储存一个序列的一维数组，DataFrame 数据结构是用于存储复杂数据的二维数据结构,本文来详细说明一下2023-07-07
这篇文章主要为大家详细介绍了python决策树之CART分类回归树，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-12-12
python Jieba分词处理详解【模式，词库的添加、删除，自定义词库，失败处理等】
这篇文章主要介绍了python Jieba分词处理,结合实例形式详细分析了python 使用jieba分词的模式，词库的添加、删除，自定义词库，失败处理等相关操作技巧,需要的朋友可以参考下2023-07-07
Python 如何利用pandas和matplotlib绘制饼图
这篇文章主要介绍了Python 如何利用pandas和matplotlib绘制饼图,代码使用了Pandas和Matplotlib库来绘制店铺销售数量占比的饼图,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了Python类及获取对象属性方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
这篇文章主要给大家介绍了关于Python面向对象基础入门之编码细节与注意事项的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2018-12-12
最新评论