logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

python中.item()函数的作用

作用:取出 单元素张量 的元素值并返回该值,保持原元素类型不变。,即:原张量元素为整形,则返回整形,原张量元素为浮点型则返回浮点型。一般运用在 神经网络 中用 loss.item()取loss值。若代码中所有的 loss值 都直接用 loss 表示的,结果就是每次迭代,空间占用就会增加,直到cpu或者gup爆炸。......

#python#深度学习#神经网络
暂无文章信息