batchnorm2d参数 torch_Pytorch-nn.BatchNorm2d()

Pytorch官方文档：

测试代码：

转自：https://blog.csdn.net/tmk_01/article/details/80679549

import torch

import torch.nn as nn

m = nn.BatchNorm2d(2,affine=True) #weight(gamma)和bias(beta)将被使用

input = torch.randn(1,2,3,4)

output = m(input)

print(“输入图片：”)

print(input)

print(“归一化权重(公式中的gamma)：”)

print(m.weight)

print(“归一化偏置(公式中的beta)：”)

print(m.bias)

print(“归一化的输出：”)

print(output)

print(“输出的尺度：”)

print(output.size())

# i = torch.randn(1,1,2)

print(“输入的第一个维度：”)

print(input[0][0])

firstDimenMean = torch.Tensor.mean(input[0][0])

firstDimenVar= torch.Tensor.var(input[0][0],False) #Bessel’s Correction贝塞尔校正不会被使用

print(m.eps)

print