torch.mtia.memory_stats

torch.mtia.memory_stats(device=None)

回傳指定裝置的 MTIA 記憶體配置器統計資料的字典。

參數

device (torch.device, str, 或 int, 選擇性) – 給定當前裝置的統計資料，如果 device 為 None (預設值)，則使用 current_device()。

回傳類型

Dict[str, Any]