torch.mtia.memory_stats¶ torch.mtia.memory_stats(device=None)[source][source]¶ 返回给定设备的 MTIA 内存分配器统计信息的字典。 参数 device (torch.device, str, 或 int, 可选) – 如果 device 为 None (默认),则返回当前设备(由 current_device() 指定)的统计信息。 返回类型 dict[str, Any]