Pytorch/Python中item()的用法

前言

在使用Pytorch训练模型时，用到python中的item()函数，如：

train_loss += loss.item()

现对item()函数用法做出总结。item()函数的作用是从包含单个元素的张量中取出该元素值，并保持该元素的类型不变。,即：该元素为整形，则返回整形，该元素为浮点型，则返回浮点型。官网解释如下：

Pytorch官网：https://pytorch.org/docs/stable/tensors.html?highlight=item#torch.Tensor.item

实验

做个测试：

import torchx = torch.randn(2, 2)print(x)
print(x[0,0])
print(x[0,0].item())

Output:

tensor([[-0.1405,  2.4767],[-0.6847,  0.0057]])
tensor(-0.1405)
-0.14052967727184296

总结

计算loss或者accuracy时，经常使用item()函数，而不是直接取对应的元素x[i,j]。
item()函数取值时，保持该元素的类型不变。

Pytorch/Python中item()的用法相关推荐

python代码大全表解释-python中的字典用法大全的代码
如下代码是关于python中的字典用法大全的代码. #!/usr/bin/env python # # [SNIPPET_NAME: Dictionaries 101] # [SNIPPET_CATE ...
python代码大全-python中的字典用法大全的代码
如下代码是关于python中的字典用法大全的代码. #!/usr/bin/env python # # [SNIPPET_NAME: Dictionaries 101] # [SNIPPET_CATE ...
python 中 del 的用法
python 中 del 的用法用法一 >>> a = [1, "two", 3, "four"] >>> del a[0 ...
python中with是什么意思,Python中With的用法
在看Dive Into Python中有关描述文件读写那章节的时候,看到了有关with的用法,查阅下相关资料,记录下来,以备后用. 官方的reference上有关with statement是这样说的 ...
Python中的else用法
Python中的else用法 1. 和if搭配使用 1.1 多分支结构 """多分支结构,多选一执行从键盘录入一个整数成绩90-100 A80-89 B70-79 C6 ...
五分钟理解yield在python中的简单用法，让你不再迷惑
很多同学无论是在学习python还是使用python的过程中,都会遇到yield关键字,这个让人头大的问题,今天,就给大家分享一下我自学yield的心路历程基本概念: (1)在 Python 中,使 ...
format函数python的顺序_[转载] Python中format函数用法
Python中format函数用法 format优点 format是python2.6新增的格式化字符串的方法,相对于老版的%格式方法,它有很多优点. 1.不需要理会数据类型的问题,在%方法中%s只能 ...
python教程实例-Python中函数的用法实例教程
本文以数值计算为例讲述了Python中函数的用法,分享给大家供大家参考借鉴之用.具体如下: 我们都知道圆的面积计算公式为: S = πr2 当我们知道半径r的值时,就可以根据公式计算出面积.假设我们需 ...
详解python中的用法_详解python中*号的用法
1.表示乘号 2.表示倍数,例如: def T(msg,time=1): print((msg+' ')*time) T('hi',3) 打印结果(打印3次): hi hi hi 3.单个 * (1) ...

Pytorch/Python中item()的用法

前言

实验

总结

Pytorch/Python中item()的用法相关推荐

最新文章

热门文章