一、问题源头
定位:print(np.array(str_reparametrize).shape)
二、原因
numpy不能读取CUDA tensor 需要将它转化为 CPU tensor。
三、解决方案
转换成CPU tenor后即可
本文采用
print(str_reparametrize.cuda().data.cpu().numpy())
四、建议
Pytorch代码运行在cpu中,原来的写是对的。
用GPU中代码运行,因为numpy在cuda中没有这种表达,需要将cuda中的数据转换到cpu中,再去使用numpy。