Implement TensorFromScalar as a pass-through

Author: rlouf

pytensor/link/jax/dispatch/tensor_basic.py

@@ -182,7 +182,7 @@ def makevector(*x):
 @jax_funcify.register(TensorFromScalar)
 def jax_funcify_TensorFromScalar(op, **kwargs):
     def tensor_from_scalar(x):
-        return jnp.array(x)
+        return x
 
     return tensor_from_scalar