mars.tensor.broadcast_arrays

mars.tensor.broadcast_arrays(*args, **kwargs)

对任何数量的数组进行广播。

Parameters

*args (array_likes) – 要广播的张量。

Returns

广播

Return type

list 的张量

示例

>>> import mars.tensor as mt

>>> x = mt.array([[1,2,3]])
>>> y = mt.array([[1],[2],[3]])
>>> mt.broadcast_arrays(x, y).execute()
[array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]]), array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])]