torch.accelerator.memory.reset_accumulated_memory_stats

torch.accelerator.memory.reset_accumulated_memory_stats(device_index=None, /)

重置指定裝置索引的當前 accelerator 記憶體分配器跟蹤的“累積”（歷史）統計資訊。

引數

device_index (torch.device, str, int, optional) – 要定位的裝置索引。如果未給出，則預設使用 torch.accelerator.current_device_index()。如果提供了 torch.device 或 str，則其型別必須與當前的 accelerator 裝置型別匹配。

注意

如果當前 accelerator 的記憶體分配器尚未初始化，則此函式將不執行任何操作。