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。