numpy.broadcast_shapes#

numpy.broadcast_shapes(*args)[source]#

将输入形状广播到单个形状。

在此处详细了解广播.

版本 1.20.0 中的新功能。

参数:
*args整数元组或整数

要相互广播的形状。

返回值:
元组

广播后的形状。

引发:
ValueError

如果形状不兼容,并且不能根据 NumPy 的广播规则进行广播。

示例

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