    def _infer_shapes(self, x: Tensor) -> List[IntVar]:
        """Infers shapes for permute210.

        Parameters
        ----------
        x : Tensor

        Returns
        ------
        List[IntVar]
            Inferred output 3d tensor with input shape.
            Because its a permute210 operation,
            Out.d0=In.d2, Out.d2=In.d0.
        """
        x_shape = x._attrs["shape"]
        return [x_shape[2], x_shape[1], x_shape[0]]
Infers shapes for permute210.

Parameters
----------
x : Tensor

Returns
------
List[IntVar]
Inferred output 3d tensor with input shape.
Because its a permute210 operation,
Out.d0=In.d2, Out.d2=In.d0.
python/aitemplate/compiler/ops/tensor/permute210.py