評價此頁

StatefulSymbolicContext#

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

透過 symbolic_context(由 Source:Symbol 的快取決定)在 create_symbolic_sizes_strides_storage_offset 中建立符號。快取命中會重用已儲存的符號,而快取未命中則會寫入此快取。

此類的行為類似於 StatelessSymbolicContext,不同之處在於快取會覆蓋其他值——如果發生快取命中,將不會讀取 dynamic_sizes 和 constraint_sizes。

快取所有者有責任維護快取與不同 shape_env 的生命週期,包括清除等。