torch.mtia.memory.memory\_stats =============================== .. currentmodule:: torch.mtia.memory .. autofunction:: memory_stats