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

        x_shape = x._attrs["shape"]
        return [x_shape[0], x_shape[2], x_shape[1], x_shape[3]]
Infers shapes for permute0213.
python/aitemplate/compiler/ops/tensor/permute0213.py