大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。
Jetbrains全系列IDE稳定放心使用
cpu tensor转numpy:
#假定a为tensor
a.numpy()
gpu tensor转numpy:
gpu下的tensor不能直接转numpy,需要先转到cpu tensor后再转为numpy
a.cpu().numpy()
注:若tensor带有梯度,以上述方式转换时会报错:
RuntimeError: Can’t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
可以改用detach().numpy()方式
a.detach().numpy()
a.cpu().detach().numpy()
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/180209.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...