美文网首页
PyTorch中.item()用法

PyTorch中.item()用法

作者: geekboys | 来源:发表于2020-03-07 23:52 被阅读0次

PyTorch有个小的知识点需要我们注意就是x.item()用法。

官方文档解释.item()用法是:一个元素张量可以用x.item()得到元素值,我理解的就是一个是张量,一个是元素。

我们也可以分别将两个值打印出来看看,首先是tensor值:


tensor

当我们获取该tensor的item值时获得值为:


tensor.item()
从上面的两张图可以看出tensor和tensor.item()的区别

相关文章

网友评论

      本文标题:PyTorch中.item()用法

      本文链接:https://www.haomeiwen.com/subject/flzkdhtx.html