fix(ivy): Resolve shape mismatch bug in the return of the torch backe…

…nd of ivy.median
ivy-llc · Aug 31, 2023 · dc8ccbc · dc8ccbc
1 parent 79ab5c8
commit dc8ccbc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ivy/functional/backends/torch/experimental/statistical.py b/ivy/functional/backends/torch/experimental/statistical.py
@@ -159,11 +159,11 @@ def median(
         interpolation="midpoint",
     )
     if input.dtype in [torch.int64, torch.float64]:
-        ret = torch.asarray(ret, dtype=torch.float64)
+        ret = ret.to(torch.float64)
     elif input.dtype in [torch.float16, torch.bfloat16]:
-        ret = torch.asarray(ret, dtype=input.dtype)
+        ret = ret.to(input.dtype)
     else:
-        ret = torch.asarray(ret, dtype=torch.float32)
+        ret = ret.to(torch.float32)
     return ret