網站首頁 編程語言 正文
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
相關推薦
- 2021-12-10 Android?Activity活動頁面跳轉與頁面傳值_Android
- 2022-10-01 React?hooks?useState異步問題及解決_React
- 2022-08-14 Redis+Caffeine兩級緩存的實現_Redis
- 2022-08-16 在WPF中使用Interaction.Triggers_C#教程
- 2022-11-21 小白也能看懂的Redis遍歷鍵和數據庫管理詳解_Redis
- 2023-11-18 數據處理使用Python提取String、字符串中的數字
- 2022-05-17 bat批處理之字符串操作的實現_DOS/BAT
- 2022-10-17 C++智能指針模板應用詳細介紹_C 語言
- 最近更新
-
- 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同步修改后的遠程分支