评价此页

torch.cuda.memory.memory_stats_as_nested_dict#

torch.cuda.memory.memory_stats_as_nested_dict(device=None)[source]#

以嵌套字典的形式返回 memory_stats() 的结果。

返回类型

dict[str, Any]