mindspore.numpy.broadcast_arrays

查看源文件
mindspore.numpy.broadcast_arrays(*args)[源代码]

将任意数量的数组广播到共同的shape。

说明

不支持Numpy的 subok 参数。在图模式下,返回的不是Tensor列表,而是Tensor的tuple。

参数:
  • *args (Tensor) - 要进行广播的数组。

返回:

Tensor列表。

异常:
  • ValueError - 如果数组不能被广播。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> x = np.array([[1,2,3]])
>>> y = np.array([[4],[5]])
>>> output = np.broadcast_arrays(x, y)
>>> print(output)
[Tensor(shape=[2, 3], dtype=Int32, value=
[[1, 2, 3],
[1, 2, 3]]), Tensor(shape=[2, 3], dtype=Int32, value=
[[4, 4, 4],
[5, 5, 5]])]