numpy.broadcast_shapes — NumPy v2.2 Manual (original) (raw)

numpy.broadcast_shapes(*args)[source]#

Broadcast the input shapes into a single shape.

Learn more about broadcasting here.

New in version 1.20.0.

Parameters:

*argstuples of ints, or ints

The shapes to be broadcast against each other.

Returns:

tuple

Broadcasted shape.

Raises:

ValueError

If the shapes are not compatible and cannot be broadcast according to NumPy’s broadcasting rules.

Examples

import numpy as np np.broadcast_shapes((1, 2), (3, 1), (3, 2)) (3, 2)

np.broadcast_shapes((6, 7), (5, 6, 1), (7,), (5, 1, 7)) (5, 6, 7)