torch.cuda.memory.memory_stats_as_nested_dict

torch.cuda.memory.memory_stats_as_nested_dict(device=None) [исходный код]

Возвращает результат memory_stats() в виде вложенного словаря.

Тип возвращаемого значения:

dict[str, Any]