大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。
Jetbrains全系列IDE稳定放心使用
解决方法
转numpy时使用Tensor.detach().numpy()
:
a = torch.ones(5)
b = a.detach().numpy()
print(b)
问题解析
当计算中的tensor
转换时,由于它带梯度值时,因此不能直接转为numpy
格式,所以最好不论如何都调用一下.detach().numpy()
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/180334.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...