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)