jax.lax.broadcast_shapes# jax.lax.broadcast_shapes(*shapes)[source]# Returns the shape that results from NumPy broadcasting of shapes.