torch.mtia.memory

創建於：2024 年 12 月 09 日 | 最後更新於：2025 年 07 月 28 日

MTIA 後端是獨立於主程式碼庫實現的，這裡僅定義了介面。

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

memory_stats	返回給定裝置的 MTIA 記憶體分配器統計資訊的字典。
memory_allocated	返回給定裝置上由張量佔用的當前 MTIA 記憶體（以位元組為單位）。