Shortcuts

DimDynamic

class torch.fx.experimental.symbolic_shapes.DimDynamic(value)[源代码]

控制如何为一个维度执行符号分配。 将其默认设置为DYNAMIC总是安全的,但策略DUCK和STATIC可以提高跟踪时间和编译时间性能,因为它们减少了分配的符号数量,并通常使您的图更加静态。

注意:如果我们注意到您已对该维度应用了约束,我们将为了简化而强制将其设置为DYNAMIC。

DimDynamic 由多种更高级别的用户体验功能控制。 目前:

  • In eager mode, the default policy is DUCK.
    • 默认值已更改为 STATIC,并启用了 assume_static_by_default。

    • 如果标记了mark_dynamic_dim,则单个维度会被标记为DYNAMIC。

  • In export mode, the default policy is STATIC.
    • 如果将某个维度标记为dynamic_dim,则该维度会被标记为DYNAMIC。

优云智算