fix: convert src to tensor in torch frontend scatter_

ivy-llc · Jul 4, 2024 · 7b46175 · 7b46175
1 parent 787b7ba
commit 7b46175
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py
@@ -1842,6 +1842,9 @@ def scatter_add_(self, dim, index, src):
         {"2.2 and below": ("float32", "float64", "int32", "int64")}, "torch"
     )
     def scatter_(self, dim, index, src, *, reduce=None):
+        if not isinstance(src, torch_frontend.Tensor):
+            src = torch_frontend.tensor(src, dtype=self.dtype)
+
         if reduce is None:
             reduce = "replace"
         else: