    def __call__(self, x: Tensor) -> Tensor:
        """
        Parameters
        ----------
        x : Tensor

        Returns
        -------
        Tensor
            Generate output tensors of function calls.
            In permute102, its a 3d tensor with d1,d0,d2 of
            input Tensor.
        """
        self._attrs["inputs"] = [x]
        self._set_depth()
        output_shape = self._infer_shapes(x)
        output = Tensor(output_shape, src_ops={self})
        output._attrs["dtype"] = x.dtype()
        self._attrs["outputs"] = [output]
        return output
Parameters
----------
x : Tensor

Returns
-------
Tensor
Generate output tensors of function calls.
In permute102, its a 3d tensor with d1,d0,d2 of
input Tensor.
python/aitemplate/compiler/ops/tensor/permute102.py