item()和items()的用法详解与区别
item()
item()的作用是取出单元素张量的元素值并返回该值,保持该元素类型不变。
听起来和使用索引来取值的作用好像一样,接下来我们看一看使用两种方法取元素值的区别:
首先定义一个张量:
1、直接使用索引取值:
2、使用item()取出
由此可以看出使用item()函数取出的元素值的精度更高,所以在求损失函数等时我们一般用item()
items()
items()的作用是把字典中的每对key和value组成一个元组,并把这些元祖放在列表中返回。
举个例子:
Albert_Roberts: 讲的很清楚
weixin_48416499: 神一样的人
恒怡爱: 这个算子非常耗时,有替代函数吗。。
data_cabbage_: 博主请问训练集测试集分别2.5那里,测试集取整变成3是这个函数的默认规定吗?这函数默认测试集优先取整?
Antrn: 还得看评论区老哥