fix: handle dtype casting for masked_fill in torch frontend (#28740)

ivy-llc · Apr 30, 2024 · 6e7807c · 6e7807c
1 parent 65a5363
commit 6e7807c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py
@@ -1062,8 +1062,9 @@ def acosh(self):
         return torch_frontend.acosh(self)
 
     def masked_fill(self, mask, value):
+        dtype = ivy.as_native_dtype(self.dtype)
         return torch_frontend.tensor(
-            torch_frontend.where(mask, value, self), dtype=self.dtype
+            ivy.astype(torch_frontend.where(mask, value, self), dtype)
         )
 
     def masked_fill_(self, mask, value):