捷徑

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:Symbol 快取給定的 symbolic_context 判斷,在 create_symbolic_sizes_strides_storage_offset 中建立符號。 快取命中將重複使用儲存的符號,而快取未命中將寫入此快取。

它的行為類似於 StatelessSymbolicContext,但快取會取代其他值 - 如果快取命中,則不會讀取 dynamic_sizes 和 constraint_sizes。

快取擁有者有責任維護快取的生命週期,使其與不同的 shape_envs 相關,進行清除等操作。

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得針對初學者和進階開發者的深入教學課程

檢視教學

資源

尋找開發資源並取得您問題的解答

檢視資源