torch.cuda.memory.memory\_stats\_as\_nested\_dict ================================================= .. currentmodule:: torch.cuda.memory .. autofunction:: memory_stats_as_nested_dict