快捷方式

DimDynamic

class torch.fx.experimental.symbolic_shapes.DimDynamic(value)[原始檔][原始檔]

控制如何對維度執行符號分配。將其預設為 DYNAMIC 始終是可靠的,但 DUCK 和 STATIC 策略可以帶來更好的跟蹤時和編譯時效能,因為它們減少了分配的符號數量,並通常使你的圖更具靜態性。

注意:如果我們發現你已對維度應用了約束,為簡單起見,我們會強制將其設定為 DYNAMIC。

DimDynamic 由多種更高級別的使用者體驗特性控制。目前

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

    • 如果你使用 mark_dynamic_dim 標記,則單個維度會被標記為 DYNAMIC。

  • 在 export 模式下,預設策略是 STATIC。
    • 如果你在傳遞給 export 的 dynamic_shapes 中指定,則單個維度會被標記為 DYNAMIC。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取適合初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源