Rate this Page
★ ★ ★ ★ ★

torch.mtia.snapshot#

torch.mtia.snapshot()[source]#

Return a dictionary of MTIA memory allocator history

Return type:

dict[str, Any]

On this page
Show Source
PyTorch Libraries