【python报错】Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.[]

脚本报错

RuntimeError:Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.

定位出错语句

enc, dec, z, gamma = self.dagmm(input_data)
            
            img_test = dec[27:36]
            plt.figure()
            for i in range(1, 9):
                plt.subplot(1, 9, i)
                plt.imshow(img_test[i - 1])#有问题
            plt.show()

分析问题：
脚本报错是在：plt.imshow(img_test[i - 1])，根据错误提示，Can’t call numpy() on Variable that requires grad. Use var.detach().numpy() ，意思是img_test不是数组格式，我将重新运行脚本发现是tensor,故需将tensor转numpy
解决方案
数据类型是tensor,将其转化成numpy格式就好用了

dec = dec.detach().numpy() 
img_test = dec.reshape(256, 28, 28)

注意：tensor 不能直接转numpy
dec = dec.numpy() 脚本还是会报错，
按报错的提示改成:
dec = dec.detach().numpy()