Created by: CHFsky
运行到此处都会内存爆掉,排查发现ln_mean没问题,是var(input,dim=[1, 2, 3],keep_dim=True)的问题。
下图为var函数,不论我自己写的还是别人写的var函数都会炸掉,不知为何,下图是别人写的,求解答