| 12345678910111213141516171819 |
- 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
- 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)
|
