torch.mtia.memory.memory_allocated

torch.mtia.memory.memory_allocated(device=None) [source]

Возвращает текущий объём памяти MTIA, занятый тензорами, в байтах для указанного устройства.

Параметры:

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

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

int