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