jax.lax.broadcast_shapes

jax.lax.broadcast_shapes#

jax.lax.broadcast_shapes(*shapes)[source]#

Returns the shape that results from NumPy broadcasting of shapes.