torch.nn.BatchNorm2d()怎么使用

这篇文章主要讲解了“torch.nn.BatchNorm2d()怎么使用”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“torch.nn.BatchNorm2d()怎么使用”吧！

实验代码展示:

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000002298027D330>>>>>>> from torch import nn>>> model = nn.BatchNorm2d(100)>>>>>> data_in = torch.randn(20,100,35,45)>>> data_in.shape
torch.Size([20, 100, 35, 45])>>>>>> data_out = model(data_in)>>> data_out.shape
torch.Size([20, 100, 35, 45])>>>>>> model = nn.BatchNorm2d(120)>>>>>>>>>>>>

感谢各位的阅读，以上就是“torch.nn.BatchNorm2d()怎么使用”的内容了，经过本文的学习后，相信大家对torch.nn.BatchNorm2d()怎么使用这一问题有了更深刻的体会，具体使用情况还需要大家实践验证。这里是亿速云，小编将为大家推送更多相关知识点的文章，欢迎关注！