This commit is contained in:
mcDandy
2026-01-20 22:03:25 +01:00
parent b5c8541a48
commit 86ed9640cb
+1 -1
View File
@@ -28,7 +28,7 @@ class UnifiedMathVisitor(MathExprVisitor):
return val.contiguous()
if self._is_list(val):
return torch.tensor(val, device=self.device)
return torch.brodcast(torch.tensor(val, device=self.device), self.shape).contiguous()
return torch.broadcast_to(torch.tensor(val, device=self.device), self.shape).contiguous()
def _bin_op(self, a, b, torch_op, scalar_op):
"""