torch.cuda.memory\_snapshot =========================== .. currentmodule:: torch.cuda .. autofunction:: memory_snapshot