fix (torch frontend)(tensor.py): fixing torch.Tensor.masked_scatter…

… where we were incorrectly broadcasting the input tensor as well.
ivy-llc · Oct 10, 2024 · 11a1c1a · 11a1c1a
1 parent e7c2e4d
commit 11a1c1a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py
@@ -1140,7 +1140,6 @@ def masked_select(self, mask):
         return torch_frontend.masked_select(self, mask)
 
     def masked_scatter(self, mask, source):
-        self = torch_frontend.broadcast_to(self, source.shape)
         mask = torch_frontend.broadcast_to(mask, self.shape)
         flat_self = torch_frontend.flatten(self.clone())
         flat_mask = torch_frontend.flatten(mask)
@@ -1150,7 +1149,6 @@ def masked_scatter(self, mask, source):
         return flat_self.reshape(self.shape)
 
     def masked_scatter_(self, mask, source):
-        self = torch_frontend.broadcast_to(self, source.shape)
         mask = torch_frontend.broadcast_to(mask, self.shape)
         flat_self = torch_frontend.flatten(self.clone())
         flat_mask = torch_frontend.flatten(mask)