Shortcuts

SubclassSymbolicContext

class torch.fx.experimental.symbolic_shapes.SubclassSymbolicContext(dynamic_sizes, constraint_sizes=None, view_base_context=None, tensor_source=None, shape_env_to_source_to_symbol_cache=None, inner_contexts=None)[source]

The correct symbolic context for a given inner tensor of a traceable tensor subclass may differ from that of the outer symbolic context. This structure allows for this flexibility, with inner symbolic contexts mapped via attr -> symbolic context.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources