torch.mtia.memory.memory_stats

torch.mtia.memory.memory_stats(device=None)

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

参数

device (torch.device, str, 或 int, 可选) – 当前设备的统计信息，如果 device 为 None（默认），则由 current_device() 给出。

返回类型

Dict[str, Any]