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

tuple