pytorch中.data()与.item

pytorch中.data()与.item,第1张

.data:
把变量设置成tensor类型,requires_grad设置为Flase
.item()
把tensor转为python的float类型

欢迎分享,转载请注明来源:内存溢出

原文地址: http://www.outofmemory.cn/langs/869156.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-13
下一篇 2022-05-13

发表评论

登录后才能评论

评论列表(0条)

保存