broadcast_tensors

paddle. broadcast_tensors ( inputs, name=None ) [源代码]

根据Broadcast规范对一组输入 inputs 进行Broadcast操作 输入应符合Broadcast规范

注解

如想了解更多Broadcasting内容，请参见 广播 (broadcasting) 。

参数

  • inputs (list(Tensor)|tuple(Tensor)) - 一组输入Tensor，数据类型为：bool、float32、float64、int32或int64。所有的输入Tensor均需要满足rank <= 5。

  • name (str，可选) - 操作的名称(可选，默认值为None）。更多信息请参见 Name 。

返回

list(Tensor) ，一组Broadcast后的 Tensor，其顺序与 input 一一对应。

代码示例

import paddle

x1 = paddle.rand([1, 2, 3, 4]).astype('float32')
x2 = paddle.rand([1, 2, 1, 4]).astype('float32')
x3 = paddle.rand([1, 1, 3, 1]).astype('float32')

out1, out2, out3 = paddle.broadcast_tensors(input=[x1, x2, x3])
# out1, out2, out3: 分别对应x1, x2, x3 Broadcast的结果，其形状均为 [1,2,3,4]