关于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常用模块(math itertools functools sys
这篇文章主要介绍了python常用模块之math itertools functools sys shutil的使用示例讲解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-06-06
这篇文章主要介绍了Python中input和raw_input的一点区别,它们都是用来读取控制台输入的函数,需要的朋友可以参考下2014-10-10
这篇文章主要介绍了python实现暗通道去雾算法的示例，帮助大家更好的利用python处理图像，感兴趣的朋友可以了解下2020-09-09
在日常工作中,我们经常需要从PDF文档中提取文本内容,虽然市面上有不少相关工具,但它们要么功能过于复杂,要么使用不够方便,本文将介绍如何使用Python开发一个简单实用的PDF文本提取工具,需要的朋友可以参考下2025-02-02
这篇文章主要介绍了用Python中的字典来处理索引统计的方法,字典的使用是Python学习当中的基础知识,本文则是相关的一个小实践,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了Pytorch实现WGAN用于动漫头像生成，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
端午节快乐，今天我将为大家带来一篇有关端午节的编程文章，希望能够为大家献上一份小小的惊喜，我们将会使用Python来实现一个与端午粽子相关的小应用程序，在本文中，我将会介绍如何用Python代码制做一个“粽子拆解器”，感兴趣的小伙伴欢迎阅读2023-06-06
python flask sqlalchemy连接数据库流程介绍
这篇文章主要介绍了python flask sqlalchemy连接数据库流程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-09-09
python3 pandas 读取MySQL数据和插入的实例
下面小编就为大家分享一篇python3 pandas 读取MySQL数据和插入的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
Flask是一个使用 Python 编写的轻量级 Web 应用框架。其 WSGI 工具箱采用 Werkzeug ，模板引擎则使用 Jinja2 。Flask使用 BSD 授权。Flask也被称为 “microframework” ，因为它使用简单的核心，用 extension 增加其他功能，Flask中Cookie和Session有什么区别呢2022-10-10
最新评论