Shortcuts

StatefulSymbolicContext

class torch.fx.experimental.symbolic_shapes.StatefulSymbolicContext(dynamic_sizes, constraint_sizes=None, view_base_context=None, tensor_source=None, shape_env_to_source_to_symbol_cache=None)[源代码]

create_symbolic_sizes_strides_storage_offset 中通过符号上下文确定创建符号,如由源:符号的缓存给出。缓存命中将重用存储的符号,缓存未命中将写入此缓存。

这表现得像 StatelessSymbolicContext,除了缓存取代了其他值 - 如果缓存命中,dynamic_sizes 和 constraint_sizes 将不会被读取。

缓存所有者的责任是维护缓存的生命周期,包括不同shape_envs、清除等。

优云智算