DimDynamic

class torch.fx.experimental.symbolic_shapes.DimDynamic(value) [исходный код]

Определяет, как выполнять выделение символа для измерения. По умолчанию всегда безопасно задавать значение DYNAMIC, но политики DUCK и STATIC могут повысить производительность трассировки и компиляции, поскольку они уменьшают количество выделяемых символов и в целом делают граф более статичным.

Примечание: если мы обнаружим, что для измерения задано ограничение, мы принудительно установим для него значение DYNAMIC для простоты.

Поведение DimDynamic определяется различными высокоуровневыми функциями взаимодействия с пользователем. В настоящее время:

• В eager-режиме политикой по умолчанию является DUCK.

  • Значение по умолчанию меняется на STATIC при использовании assume_static_by_default.
  • Отдельное измерение помечается как DYNAMIC, если вызвать mark_dynamic_dim.

• В режиме экспорта политикой по умолчанию является STATIC.

  • Отдельное измерение помечается как DYNAMIC, если указать его в dynamic_shapes, передаваемом в export.