python中.item

python中.item,第1张

作用:取出 单元素张量 的元素值并返回该值,保持原元素类型不变。,即:原张量元素为整形,则返回整形,原张量元素为浮点型则返回浮点型。

一般运用在 神经网络 中用 loss.item()取loss值。

若代码中所有的 loss值 都直接用 loss 表示的,结果就是每次迭代,空间占用就会增加,直到cpu或者gup爆炸。

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

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

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

发表评论

登录后才能评论

评论列表(0条)

保存