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 的生命周期、清除等。