jax.numpy.broadcast_shapes

jax.numpy.broadcast_shapes(*shapes: Sequence[int]) → tuple[int, ...]

jax.numpy.broadcast_shapes(*shapes: Sequence[int | Tracer]) → tuple[int | Tracer, ...]

Broadcast the input shapes into a single shape.

LAX-backend implementation of numpy.broadcast_shapes().

Original docstring below.

Learn more about broadcasting here.

Added in version 1.20.0.

Returns:

Broadcasted shape.

Return type:

tuple