1、item()
item()的作用是取出单元素张量(即只取一个元素才不会报错)的元素值并返回该值,保持该元素类型不变。
import torch
a = torch.randn((5))
print(a)
# 1、直接使用索引取值
print(a[1])
# 2、使用item()取值
print(a[1].item())
# 输出结果
# tensor([-0.6861, -0.6912, 1.3144, -0.8755, -2.1023])
# tensor(-0.6912)
# -0.6911846399307251
由此可以看出使用item()函数取出的元素值的精度更高,所以在求损失函数等时我们一般用item()
2、items()
items()的作用是把字典中的每对key和value组成一个元组,并把这些元祖放在列表中返回。
a = {"a1":2,"a2":3,"a3":5}
print(a.items())
# 输出结果
# dict_items([('a1', 2), ('a2', 3), ('a3', 5)])