Readded

pasqal-io · Oct 16, 2023 · b7faa6a · b7faa6a
1 parent 9a0bf02
commit b7faa6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qadence/ml_tools/models.py b/qadence/ml_tools/models.py
@@ -181,7 +181,7 @@ def _transform_x(self, x: dict[str, torch.Tensor] | Tensor) -> dict[str, Tensor]
 
         else:
             assert isinstance(self.model, torch.nn.Module) and isinstance(x, Tensor)
-            return self._input_scaling * x + self._input_shifting
+            return self._input_scaling * (x + self._input_shifting)
 
     def forward(self, x: dict[str, Tensor] | Tensor, *args: Any, **kwargs: Any) -> Tensor:
         y = self.model(self._transform_x(x), *args, **kwargs)