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, shape_ids=None, unbacked_bounds=None, *, tensor_source, shape_env_to_source_to_symbol_cache=<factory>, excluded_sizes=None) [исходный код]

Создаёт символы в _create_symbolic_sizes_strides_storage_offset посредством определения symbolic_context, заданного кэшем Source:Symbol. При попадании в кэш используется сохранённый символ, а при промахе символ записывается в этот кэш.

Работает подобно StatelessSymbolicContext, за исключением того, что кэш имеет приоритет над остальными значениями: при попадании в кэш значения dynamic_sizes и constraint_sizes считываться не будут.

Ответственность за управление жизненным циклом кэша с учётом различных shape_envs, его очистки и т. д. лежит на владельце кэша.