網站首頁 編程語言 正文
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
相關推薦
- 2022-07-18 linux引導和計劃任務
- 2022-08-27 C#實現截圖幫助類_C#教程
- 2022-10-20 Swift初始化器與可選鏈的使用方法介紹_Swift
- 2023-03-04 Go語言實現分布式鎖_Golang
- 2023-05-19 Kotlin?Suspend掛起函數的使用詳解_Android
- 2023-07-03 Python?Thread虛假喚醒概念與防范詳解_python
- 2022-04-16 C語言實現順序循環隊列實例_C 語言
- 2022-05-01 教你利用python如何讀取txt中的數據_python
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支