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