jax.numpy.broadcast_shapes

Warning

This page was created from a pull request (#9655).

jax.numpy.broadcast_shapes¶

jax.numpy.broadcast_shapes(*shapes)[source]¶

Broadcast the input shapes into a single shape.

LAX-backend implementation of broadcast_shapes().

Original docstring below.

Learn more about broadcasting here.

New in version 1.20.0.

Returns

Broadcasted shape.

Return type

tuple