pytorch Tensor转numpy并解决RuntimeError: Can‘t call numpy() on Tensor that requires grad.报错
  TEZNKK3IfmPf 2023年11月14日 50 0

解决方法

转numpy时使用Tensor.detach().numpy()

a = torch.ones(5)
b = a.detach().numpy()
print(b)

问题解析

当计算中的tensor转换时,由于它带梯度值时,因此不能直接转为numpy格式,所以最好不论如何都调用一下.detach().numpy()

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月14日 0

暂无评论

推荐阅读
  TEZNKK3IfmPf   2024年03月29日   61   0   0 pytorch
  I7JaHrFMuDsU   2024年04月26日   32   0   0 pytorch
  TEZNKK3IfmPf   2023年11月15日   52   0   0 算法numpypython
TEZNKK3IfmPf