fix: convert scalar tensors to native python type for values in torch…

… backend put_along_axis
ivy-llc · Jul 9, 2024 · ace8434 · ace8434
1 parent cf9e5b4
commit ace8434
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ivy/functional/backends/torch/experimental/manipulation.py b/ivy/functional/backends/torch/experimental/manipulation.py
@@ -465,8 +465,8 @@ def put_along_axis(
     }
     mode = mode_mappings.get(mode, mode)
     indices = indices.to(torch.int64)
-    if not isinstance(values, torch.Tensor):
-        values = torch.tensor(values)
+    if isinstance(values, torch.Tensor) and values.dim() == 0:
+        values = values.item()
     if mode == "replace":
         return torch.scatter(arr, axis, indices, values, out=out)
     else: