fix: cast to elemwise outputs to their respective dtypes

Author: ferrine

pytensor/scalar/basic.py

@@ -1142,12 +1142,12 @@ def output_types(self, types):
 
     def perform(self, node, inputs, output_storage):
         if self.nout == 1:
-            output_storage[0][0] = self.impl(*inputs)
+            output_storage[0][0] = self.impl(*inputs).astype(node.outputs[0].dtype, copy=False)
         else:
             variables = from_return_values(self.impl(*inputs))
             assert len(variables) == len(output_storage)
-            for storage, variable in zip(output_storage, variables):
-                storage[0] = variable
+            for out, storage, variable in zip(node.outputs, output_storage, variables):
+                storage[0] = variable.astype(out.dtype, copy=False)
 
     def impl(self, *inputs):
         raise MethodNotDefined("impl", type(self), self.__class__.__name__)