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]
