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)