torch.mtia.get_device_properties

torch.mtia.get_device_properties(device=None) [исходный код]

Возвращает словарь свойств устройства MTIA

Параметры:

device (torch.device или int, необязательный) – статистика для текущего устройства, возвращаемая функцией current_device(), если device равен None (значение по умолчанию).

Тип возвращаемого значения:

dict[str, Any]