U Mh@s(dZddlZddlZddZddZdS)anThis 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. NcCs t|S)a.Get the shape of a tensor as a tensor. Args: x (Tensor): The input tensor. Returns: Tensor: A tensor of shape [len(x.shape)] containing the size of each dimension of x. Example: >>> x = torch.randn(2, 3) >>> shape_as_tensor(x) tensor([2, 3]) )torchZ_shape_as_tensor)xrF/var/www/html/venv/lib/python3.8/site-packages/torch/onnx/operators.pyshape_as_tensorsrcCs t||S)N)rZ_reshape_from_tensor)rshaperrrreshape_from_tensor_shape"sr)__doc__rZ torch.onnxrrrrrrs