RuntimeError: Can't call numpy()

2021-12-30  本文已影响0人  王叽叽的小心情

代码背景:在对计算的结果进行R-squard计算时,调用np.array(a)进行数据类型转换。

错误提示:RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.

解决方法:
当计算中的数据类型为tensor时,想要获取其具体的数值,由于它带梯度值时,不能直接转为numpy格式,所以最好不论如何都调用一下a.detach().numpy(),其中a是要转换的数据。

上一篇下一篇

猜你喜欢

热点阅读