torch.xpu.reset_accumulated_memory_stats¶
- torch.xpu.reset_accumulated_memory_stats(device=None)[原始碼][原始碼]¶
重設 XPU 記憶體分配器追蹤的「累積」（歷史）統計資訊。
詳情請參閱
memory_stats()。累積的統計數據對應於每個個別統計字典中的 “allocated” 和 “freed” 鍵。- 參數
device (torch.device 或 int 或 str, optional) – 選擇的裝置。 如果
device為None(預設值)，則傳回由current_device()給定的目前裝置的統計數據。
