日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

關于torch中tensor數據類型的轉換_python

作者:深淵為陵 ? 更新時間: 2022-12-10 編程語言

torch中tensor數據類型轉換

轉換為float32類型:

points = points.type(torch.float32)

讀取torch.Tensor數據類型

最近弄一下yolov5的detect.py,想提取一下檢測數據,記錄

a = torch.tensor(4., device='cuda:0')
print(a.item(), a.dtype, a.data, a.int())
'''
輸出:
4.0 
torch.float32 
tensor(4., device='cuda:0') 
tensor(4, device='cuda:0', dtype=torch.int32)
*********************************************
a.item()  輸出里面的數字(yolo中即代表class的下標)
a.dtype	輸出數據類型
a.data	輸出當前的數據,同print(a)
a.int() 數據類型轉換成整型
'''

torch.Tensor 類型,具體 data 為 tensor(1., device=‘cuda:0’)

跳轉看一下類型的定義(ctrl+點擊tensor)

a = torch.tensor()

可以看到定義是

def tensor(data: Any, dtype: Optional[_dtype]=None, 
device: Union[_device, str, None]=None, 
requires_grad: _bool=False) -> Tensor: ...

原文鏈接:https://blog.csdn.net/m0_46345373/article/details/123744198

欄目分類
最近更新