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。