numpy.broadcast_shapes

numpy.broadcast_shapes(*args)

将输入形状广播为单一形状.

了解更多关于广播的信息.

在 1.20.0 版本加入.

参数:

*args整数的元组,或整数

要相互广播的形状.

返回:

tuple

广播形状.

引发:

ValueError

如果形状不兼容且无法根据 NumPy 的广播规则进行广播.

参见

broadcast

broadcast_arrays

broadcast_to

示例

>>> 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)