torch.mps.current_allocated_memory

torch.mps.current_allocated_memory() [source]

Возвращает текущее занимаемое ГП памяти тензорами в байтах.

Примечание

Возвращаемый размер не включает кэшированные выделения в пулах памяти MPSAllocator.

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

int