torch.autograd.profiler.profile.key_averages

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

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

Параметры:

• group_by_input_shapes – группировать записи по паре (имя события, формы входных данных), а не только по имени события. Это полезно, чтобы увидеть, какие формы входных данных вносят наибольший вклад во время выполнения, и может помочь в оптимизациях для конкретных размеров или при выборе лучших кандидатов для квантования (то есть при подборе к линии крыши)
• group_by_stack_n – группировать по первым n записям трассировки стека
• group_by_overload_name – различать операторы по имени их перегрузки, например aten::add.Tensor
• separately (и aten::add.out будут агрегированы) –
• include_python_functions – включать в усреднение события функций Python (например, отдельные записи мест вызова Python, собранные с помощью with_stack=True). По умолчанию они исключены, поскольку часто создают вводящие в заблуждение узкие места (например, threading.py: wait), которые скрывают реальную разбивку на уровне операторов. Установите значение True, чтобы восстановить необработанное представление по отдельным местам вызова.

Возвращает:

EventList, содержащий объекты FunctionEventAvg.