評價此頁

torch.cuda.memory.memory_stats_as_nested_dict#

torch.cuda.memory.memory_stats_as_nested_dict(device=None)[原始碼]#

以巢狀字典的形式返回 memory_stats() 的結果。

返回型別

dict[str, Any]