torch.mtia.memory

MTIA 後端是在 PyTorch 程式碼樹之外實現的，這裡僅定義了介面。

此包增加了對 MTIA 中實現的裝置記憶體管理的支援。

memory_stats

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