torch.mtia.memory.memory_stats

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

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

Параметры:

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

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

dict[str, Any]