Fix bug in broadcast_to Op with Jax backend when keras variable as in…

…put (keras-team#19118) * Fix bug in broadcast_to Op with Jax backend when keras variable as input * Fix format error * convert_to_numpy chnaged to convert_to_tensor * Removed duplicate import of convert_to_tensor
Faisal-Alsrheed · Jan 31, 2024 · 7736f67 · 7736f67
1 parent 186f872
commit 7736f67
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/backend/jax/numpy.py b/keras/backend/jax/numpy.py
@@ -322,6 +322,7 @@ def average(x, axis=None, weights=None):
 
 
 def broadcast_to(x, shape):
+    x = convert_to_tensor(x)
     return jnp.broadcast_to(x, shape)