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)