`numpy:broadcast_shapes`

Signature

def broadcast_shapes(*args)

Broadcast the input shapes into a single shape.

*args : tuples of ints, or ints: The shapes to be broadcast against each other.

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

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)