torch.autograd.profiler.profile.key_averages

profile.key_averages(group_by_input_shape=False, group_by_stack_n=0) [source]

Среднее значение всех событий функций по их ключам.

Параметры

• group_by_input_shapes – группировать записи по (имя события, формы входных данных), а не только по имени события. Это полезно для того, чтобы увидеть, какие формы входных данных вносят наибольший вклад во время выполнения, и может помочь в оптимизациях, зависящих от размера, или в выборе лучших кандидатов для квантования (т.е. построения кривой крыши)
• group_by_stack_n – группировать по верхним n записям стека вызовов

Возвращает

Список событий (EventList) содержащий объекты FunctionEventAvg.