Files
pytorch/torch/onnx/operators.py
Justin Chu 88eff10722 [ONNX] Deprecate operators.py (#87798)
Deprecate `torch.onnx.operators` because it's only for backwards compatibility
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87798
Approved by: https://github.com/BowenBao
2022-10-26 20:42:06 +00:00

24 lines
753 B
Python

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
from torch.onnx import _deprecation
# 180-day deprecation period
@_deprecation.deprecated("1.14", "1.16", "use torch._shape_as_tensor")
def shape_as_tensor(x):
return torch._shape_as_tensor(x)
@_deprecation.deprecated("1.14", "1.16", "use torch._reshape_from_tensor")
def reshape_from_tensor_shape(x, shape):
return torch._reshape_from_tensor(x, shape)