add_n

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

对输入的一至多个 Tensor 求和。

Case 1:
    输入:
        input.shape = [2, 3]
        input = [[1, 2, 3],
              [4, 5, 6]]

    输出:
        output.shape = [2, 3]
        output = [[1, 2, 3],
                [4, 5, 6]]

Case 2:
    输入:
    第一个输入:
            input1.shape = [2, 3]
            input1 = [[1, 2, 3],
                  [4, 5, 6]]

    第二个输入:
            input2.shape = [2, 3]
            input2 = [[7, 8, 9],
                  [10, 11, 12]]

    输出:
        output.shape = [2, 3]
        output = [[8, 10, 12],
              [14, 16, 18]]

参数

  • inputs (Tensor|list(Tensor)) - 输入的一至多个 Tensor。如果输入了多个 Tensor,则不同 Tensor 的 shape 和数据类型应保持一致。数据类型支持:float32、float64、int32、int64。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor,输入 inputs 求和后的结果,shape 和数据类型与 inputs 一致。

代码示例

import paddle

input0 = paddle.to_tensor([[1, 2, 3], [4, 5, 6]], dtype='float32')
input1 = paddle.to_tensor([[7, 8, 9], [10, 11, 12]], dtype='float32')
output = paddle.add_n([input0, input1])
# [[8., 10., 12.],
#  [14., 16., 18.]]

使用本API的教程文档