torch.xpu.get_device_capability¶
- torch.xpu.get_device_capability(device=None)[source][source]¶
獲取裝置的 xpu 能力。
- 引數
device (torch.device or int or str, 可選) – 要返回裝置能力的裝置。如果此引數為負整數，則此函式不起作用。如果
device為None（預設值），則使用由current_device()指定的當前裝置。- 返回值
裝置的 xpu 能力字典
- 返回型別
Dict[str, Any]
