Rate this Page

torch.xpu.memory_stats_as_nested_dict#

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

Return the result of memory_stats() as a nested dictionary.

Return type

dict[str, Any]