fix: Fix ivy.permute_dims for all backends (#28009)

Co-authored-by: NripeshN <86844847+NripeshN@users.noreply.github.com>

ivy/functional/backends/jax/manipulation.py

@@ -76,6 +76,9 @@ def permute_dims(
     copy: Optional[bool] = None,
     out: Optional[JaxArray] = None,
 ) -> JaxArray:
+    if copy:
+        newarr = jnp.copy(x)
+        return jnp.transpose(newarr, axes)
     return jnp.transpose(x, axes)
 
 

ivy/functional/backends/numpy/manipulation.py

@@ -86,6 +86,9 @@ def permute_dims(
     copy: Optional[bool] = None,
     out: Optional[np.ndarray] = None,
 ) -> np.ndarray:
+    if copy:
+        newarr = np.copy(x)
+        return np.transpose(newarr, axes)
     return np.transpose(x, axes)
 
 

ivy/functional/backends/paddle/manipulation.py

@@ -108,18 +108,8 @@ def flip(
     return paddle.flip(x, axis)
 
 
-@with_supported_dtypes(
-    {
-        "2.6.0 and below": (
-            "int32",
-            "int64",
-            "float64",
-            "complex128",
-            "float32",
-            "complex64",
-            "bool",
-        )
-    },
+@with_unsupported_dtypes(
+    {"2.6.0 and below": ("uint8", "int8", "int16", "bfloat16", "float16")},
     backend_version,
 )
 def permute_dims(
@@ -130,6 +120,9 @@ def permute_dims(
     copy: Optional[bool] = None,
     out: Optional[paddle.Tensor] = None,
 ) -> paddle.Tensor:
+    if copy:
+        newarr = paddle.clone(x)
+        return paddle.transpose(newarr, axes)
     return paddle.transpose(x, axes)
 
 

ivy/functional/backends/torch/manipulation.py

@@ -78,6 +78,7 @@ def flip(
     return torch.flip(x, new_axis)
 
 
+@with_unsupported_dtypes({"2.1.2 and below": ("bfloat16", "float16")}, backend_version)
 def permute_dims(
     x: torch.Tensor,
     /,
@@ -86,6 +87,9 @@ def permute_dims(
     copy: Optional[bool] = None,
     out: Optional[torch.Tensor] = None,
 ) -> torch.Tensor:
+    if copy:
+        newarr = torch.clone(x).detach()
+        return torch.permute(newarr, axes)
     return torch.permute(x, axes)