Python numpy broadcast_shapes用法及代码示例

本文简要介绍 python 语言中 numpy.broadcast_shapes 的用法。

用法:

numpy.broadcast_shapes(*args)

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

在此处了解有关广播的更多信息。

参数：

`*args`： 整数元组或整数

要相互广播的形状。

返回：

元组

广播的形状。

抛出：

ValueError

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

例子：

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