快捷方式

DimDynamic

class torch.fx.experimental.symbolic_shapes.DimDynamic(value)[source][source]

控制如何為維度執行符號分配。預設設定為 DYNAMIC 總是合理的,但 DUCK 和 STATIC 策略可以帶來更好的追蹤時間和編譯時間效能,因為它們減少了分配的符號數量,並且通常使您的圖形更加靜態。

注意:如果我們注意到您已將約束應用於維度,為了簡化,我們將強制其為 DYNAMIC。

DimDynamic 由各種更高等級的 UX 功能控制。目前:

  • 在 eager 模式下,預設策略是 DUCK。
    • 預設值會透過 assume_static_by_default 更改為 STATIC。

    • 如果您使用 mark_dynamic_dim 標記,則個別的 dim 會被標記為 DYNAMIC。

  • 在匯出模式下,預設策略是 STATIC。
    • 如果您在傳遞給匯出的 dynamic_shapes 中指定,則個別的 dim 會被標記為 DYNAMIC。

文件

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources