评价此页

MemRecordsAcc#

class torch.autograd.profiler_util.MemRecordsAcc(mem_records)[source]#

用于访问 interval 中 mem_records 的加速结构。

in_interval(start_us, end_us)[source]#

返回给定区间内的所有记录。为保持向后兼容,函数中将 us 转换为 ns。