torch.mtia.memory_stats

torch.mtia.memory_stats(device=None)

返回给定设备的 MTIA 内存分配器统计信息的字典。

参数:

device（torch.device、str 或 int，可选）- 当前设备统计信息，由 current_device()提供，如果 device 为 None（默认）。

返回类型:

dict[str, Any]