目錄

快捷方式

torch.mtia.get_device_capability¶

torch.mtia.get_device_capability(device=None)[source][source]¶

返回給定裝置的能力，格式為 (主版本號, 次版本號) 的元組。

引數: device (torch.device 或 int, 可選) – 如果 device 為 None（預設值），則返回當前裝置（透過 current_device() 獲取）的能力統計資訊。
返回型別: tuple[int, int]

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源