快捷方式

DimDynamic

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

控制如何为维度执行符号分配。始终将此默认设置为 DYNAMIC 是安全的,但策略 DUCK 和 STATIC 可能导致更好的跟踪时间和编译时间性能,因为它们减少了分配的符号数量,通常使您的图更静态。

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

DimDynamic 由各种更高级别的 UX 功能控制。目前

  • 在急切模式下,默认策略为 DUCK。
    • 默认值使用 assume_static_by_default 更改为 STATIC。

    • 如果使用 mark_dynamic_dim 标记单个维度,则将其标记为 DYNAMIC。

  • 在导出模式下,默认策略为 STATIC。
    • 如果在传递给 export 的 dynamic_shapes 中指定单个维度,则将其标记为 DYNAMIC。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源