torch.cuda.memory.host_memory_stats_as_nested_dict

torch.cuda.memory.host_memory_stats_as_nested_dict() [исходный код]

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

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

dict[str, Any]