TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.memory_stats_as_nested_dict

tensorplay.cuda.memory_stats_as_nested_dict(device: Any | None = None) dict[str, Any][source]

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

Ask DeepWiki