DimDynamic#
- class torch.fx.experimental.symbolic_shapes.DimDynamic(value)[source]#
控制一个维度的符号分配方式。将此默认设置为DYNAMIC始终是安全的,但DUCK和STATIC策略可以带来更好的跟踪时间和编译时间性能,因为它们减少了分配的符号数量,并通常使您的图更具静态性。
注意:如果我们发现您已为该维度应用了约束,我们将出于简便性将其强制设为DYNAMIC。
DimDynamic由各种更高级别的用户体验功能控制。当前
- 在急切模式下,默认策略是DUCK。
默认策略通过assume_static_by_default更改为STATIC。
单个维度通过mark_dynamic_dim标记为DYNAMIC。
- 在导出模式下,默认策略是STATIC。
单个维度通过指定给export的dynamic_shapes来标记为DYNAMIC。