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