item()和items()的用法详解与区别

item()

item()的作用是取出单元素张量的元素值并返回该值,保持该元素类型不变。
听起来和使用索引来取值的作用好像一样,接下来我们看一看使用两种方法取元素值的区别:
首先定义一个张量:
在这里插入图片描述

1、直接使用索引取值:
在这里插入图片描述
2、使用item()取出
在这里插入图片描述
由此可以看出使用item()函数取出的元素值的精度更高,所以在求损失函数等时我们一般用item()

items()

items()的作用是把字典中的每对key和value组成一个元组,并把这些元祖放在列表中返回。
举个例子:
在这里插入图片描述

你可能感兴趣的:(python)