在pytorch模型中如何获得BatchNorm2d层的各个mean和var(平均值和方差)

NoSuchKey