python中.item()的讲解
生活随笔
收集整理的這篇文章主要介紹了
python中.item()的讲解
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
在pytorch訓練時,一般用到.item()。比如loss.item()。我們可以做個簡單測試代碼看看它的區別。
import torch x = torch.randn(2, 2) print(x) print(x[1,1]) print(x[1,1].item())結果為:
tensor([[ 0.4702, 0.5145],[-0.0682, -1.4450]]) tensor(-1.4450) -1.445029854774475可以看出是顯示精度的區別,item()返回的是一個浮點型數據,所以我們在求loss或者accuracy時,一般使用item(),而不是直接取它對應的元素x[1,1]。
有用請點個贊!!
本站所有文章均為原創,歡迎轉載,請注明文章出處:https://blog.csdn.net/weixin_45092662。百度和各類采集站皆不可信,搜索請謹慎鑒別。技術類文章一般都有時效性,本人習慣不定期對自己的博文進行修正和更新,因此請訪問出處以查看本文的最新版本。
總結
以上是生活随笔為你收集整理的python中.item()的讲解的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 响铃:金蝶的SaaS第一能保持多久?
- 下一篇: QGraphicsItem限制拖动方向和