DimDynamic¶
- class torch.fx.experimental.symbolic_shapes.DimDynamic(value)[源代码][源代码]¶
控制如何为维度执行符号分配。始终可以安全地将其默认为 DYNAMIC,但策略 DUCK 和 STATIC 可以带来更好的跟踪时间和编译时性能,因为它们减少了分配的符号数量,并通常使您的图表更静态。
注意:如果我们注意到您已对维度应用约束,为了简单起见,我们将强制其为 DYNAMIC。
DimDynamic 由各种更高级别的 UX 功能控制。目前
- 在 eager 模式下,默认策略为 DUCK。
通过 assume_static_by_default,默认值更改为 STATIC。
如果您 mark_dynamic_dim,则单个维度标记为 DYNAMIC。
- 在导出模式下,默认策略为 STATIC。
如果您在传递给 export 的 dynamic_shapes 中指定维度,则单个维度标记为 DYNAMIC。