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]