快捷方式

StatefulSymbolicContext

class torch.fx.experimental.symbolic_shapes.StatefulSymbolicContext(dynamic_sizes, dynamic_strides=None, constraint_sizes=None, constraint_strides=None, view_base_context=None, tensor_source=None, shape_env_to_source_to_symbol_cache=None)[source][source]

透過 Source:Symbol 快取確定的 symbolic_context,在 create_symbolic_sizes_strides_storage_offset 中建立符號。快取命中將重用儲存的符號,而快取未命中將寫入此快取。

其行為類似於 StatelessSymbolicContext,但快取優先於其他值 - 如果快取命中,將不會讀取 dynamic_sizes 和 constraint_sizes。

快取所有者負責維護快取相對於不同 shape_envs 的生命週期、清除等。


© 版權所有 PyTorch 貢獻者。

使用 Sphinx 構建,主題由 Read the Docs 提供。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

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

檢視資源