Implement `TensorFromScalar` as a pass-through

pytensor/link/jax/dispatch/tensor_basic.py

@@ -182,7 +182,7 @@ def jax_funcify_MakeVector(op, **kwargs):
 @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