torch.mtia.memory_stats

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

Возвращает словарь статистики распределителя памяти MTIA для указанного устройства.

Параметры:

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

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

dict[str, Any]