PyTorch有个小的知识点需要我们注意就是x.item()用法。
官方文档解释.item()用法是:一个元素张量可以用x.item()得到元素值,我理解的就是一个是张量,一个是元素。
我们也可以分别将两个值打印出来看看,首先是tensor值:
![](https://img.haomeiwen.com/i11975952/49abad065f7d99ce.png)
当我们获取该tensor的item值时获得值为:
![](https://img.haomeiwen.com/i11975952/1e25092641e39e9c.png)
从上面的两张图可以看出tensor和tensor.item()的区别
PyTorch有个小的知识点需要我们注意就是x.item()用法。
官方文档解释.item()用法是:一个元素张量可以用x.item()得到元素值,我理解的就是一个是张量,一个是元素。
我们也可以分别将两个值打印出来看看,首先是tensor值:
当我们获取该tensor的item值时获得值为:
本文标题:PyTorch中.item()用法
本文链接:https://www.haomeiwen.com/subject/flzkdhtx.html
网友评论