# 对(N, H, W)计算均值方差
mean = x1.mean(dim=).reshape(, , , )
# x1.mean(dim=1)后维度为(3,)
std = x1.std(dim=, unbiased...官方代码
x1 = x.reshape(, -1) # 对(C,H,W)计算均值方差
mean = x1.mean(dim=).reshape(, , , )
std = x1.std(dim=, unbiased...官方代码
x1 = x.reshape(, -1) # 对(H,W)计算均值方差
mean = x1.mean(dim=).reshape(, , , )
std = x1.std(dim=, unbiased...官方代码
x1 = x.reshape(,,-1) # 对(H,W)计算均值方差
mean = x1.mean(dim=).reshape(, , -1)
std = x1.std(dim=, unbiased