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