Spaces:
Sleeping
Sleeping
File size: 580 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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)
|