jax.numpy.broadcast_shapes#
- jax.numpy.broadcast_shapes(*shapes)[source]#
Broadcast the input shapes into a single shape.
LAX-backend implementation of
numpy.broadcast_shapes()
.Original docstring below.
Learn more about broadcasting here.
New in version 1.20.0.
- Returns:
Broadcasted shape.
- Return type: