快捷键

状态符号上下文 ¶

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]

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

这类似于 StatelessSymbolicContext,但缓存会覆盖其他值 - 如果缓存命中,则不会读取 dynamic_sizes 和 constraint_sizes。

维护缓存的生命周期是缓存所有者的责任,包括与不同 shape_envs 相关的操作以及清除等。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源