运行代码报错:
invalid gradient at index 0 - expected shape [] but got [1]
定位的代码段是: d_real.backward(mone)
其中mone的定义如下:
one=torch.Tensor([1])
mone=one*-1
mone=mone.cuda()
不知道是为什么,但是后来将 对应的代码段改为d_real.backward(mone.mean())后就正常运行了.
运行代码报错:
invalid gradient at index 0 - expected shape [] but got [1]
定位的代码段是: d_real.backward(mone)
其中mone的定义如下:
one=torch.Tensor([1])
mone=one*-1
mone=mone.cuda()
不知道是为什么,但是后来将 对应的代码段改为d_real.backward(mone.mean())后就正常运行了.