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。