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.