Shortcuts

SymbolicContext

class torch.onnx.SymbolicContext(params_dict, env, cur_node, onnx_block)[source]

Provides extra context for symbolic functions.

Parameters
  • params_dict (Dict[str, _C.IValue]) – Mapping from graph initializer name to IValue.

  • env (Dict[_C.Value, _C.Value]) – Mapping from Torch domain graph Value to ONNX domain graph Value.

  • cur_node (_C.Node) – Current node being converted to ONNX domain.

  • onnx_block (_C.Block) – Current ONNX block that converted nodes are being appended to.

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