評價此頁

MemRecordsAcc#

class torch.autograd.profiler_util.MemRecordsAcc(mem_records)[原始碼]#

用於在給定時間間隔內訪問 mem_records 的加速結構。

in_interval(start_us, end_us)[原始碼]#

返回給定時間間隔內的所有記錄。為了保持向後相容性,函式內會將 us 轉換為 ns。