Python-.item()的理解与使用
生活随笔
收集整理的這篇文章主要介紹了
Python-.item()的理解与使用
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
簡介
最近看到代碼中多次使用到item,于是開始學習,但是搜item出來的大多全是items()遍歷。于是終于找到一篇關于python中.item()的用法
官方手冊參考
item理解:
- 取出張量具體位置的元素元素值,并且返回的是該位置元素值的高精度值,保持原元素類型不變;必須指定位置
- 即:原張量元素為整形,則返回整形,原張量元素為浮點型則返回浮點型,etc.
使用
- 求loss,以及accuracy rate的時候一般用item(),因為獲取的值精度高
舉例
代碼
import torch x = torch.randn(2,2) print(x) # 【2,2】,張量 print(x[1,1]) # 直接獲取右下角值,張量 print(x[1,1].item()) # 獲取右下角高精度值,數值 print(x[:,1].item) # 貌似有問題,也就是獲取某個值的時候必須指定具體位置(行和列都表明) print(x) # 【2,2】 print(x[:, 0]) # 【1,2】,獲取第一列鋪成一維張量 print(x[:, 0].max()) # 【1總結
以上是生活随笔為你收集整理的Python-.item()的理解与使用的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: torch.nn.Embedding(n
- 下一篇: 单元测试系列一-为什么要写单元测试,何时