python中.item()函数的作用
作用：取出 单元素张量 的元素值并返回该值，保持原元素类型不变。,即：原张量元素为整形，则返回整形，原张量元素为浮点型则返回浮点型。一般运用在 神经网络 中用 loss.item()取loss值。若代码中所有的 loss值 都直接用 loss 表示的，结果就是每次迭代，空间占用就会增加，直到cpu或者gup爆炸。......
·
作用：取出 单元素张量 的元素值并返回该值，保持原元素类型不变。,即：
单元素的张量转为对应的int型或者float型。
一般运用在 神经网络 中用 loss.item()取loss值。
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
6
0- 0
已为社区贡献1条内容
所有评论(0)