Source code for torch.onnx.operators

r"""This file provides a location for operators that help exporting
models via onnx. E.g. shape_as_tensor and reshape_from_tensor_shape
are to make all dynamic sizes operations traceable.

NOTE: at one point these functions were implemented differently.
Since then we have implemented these directly in ATen, so this
file is kept purely for backward-compatibility.

import torch
import torch.onnx
import torch.onnx.utils

[docs]def shape_as_tensor(x): return torch._shape_as_tensor(x)
def reshape_from_tensor_shape(x, shape): return torch._reshape_from_tensor(x, shape)


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources