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