torch.xpu.get_device_capability¶
- torch.xpu.get_device_capability(device=None)[源代码]¶
获取设备的xpu能力。
- Parameters
设备 (torch.device 或 整数 或 字符串, 可选) – 返回设备能力的设备。如果此参数为负整数，则此函数为无操作。如果
device为None（默认），则使用由current_device()给出的当前设备。- Returns
设备的xpu能力字典
- Return type
Dict[str, Any]