MemRecordsAcc

class torch.autograd.profiler_util.MemRecordsAcc(mem_records)

用于在给定时间间隔内访问 mem_records 的加速结构。

in_interval(start_us, end_us)

返回给定时间间隔内的所有记录。为了保持向后兼容性，函数内会将 us 转换为 ns。