torch.mtia.memory

MTIA 後端在樹外實作，此處僅定義介面。

此套件增加了對 MTIA 中實作的裝置記憶體管理支援。

memory_stats

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