torch.mtia.memory.memory_stats

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

傳回給定裝置的 MTIA 記憶體分配器統計資訊字典。

參數

device (torch.device, str, or int, optional) – 如果 device 為 None (預設)，則給定 current_device() 的目前裝置的統計資訊。

傳回類型

Dict[str, Any]