sequence_concat¶
该 OP 的输入只能是 LoDTensor,如果您需要处理的输入是 Tensor 类型,请使用 paddle.concat 。
该 OP 仅支持 LoDTensor,通过 LoDTensor 的 LoD 信息将输入的多个 LoDTensor 进行连接(concat),输出连接后的 LoDTensor。
input 是由多个 LoDTensor 组成的 list:
input = [x1, x2]
其中:
x1.lod = [[0, 3, 5]]
x1.data = [[1], [2], [3], [4], [5]]
x1.shape = [5, 1]
x2.lod = [[0, 2, 4]]
x2.data = [[6], [7], [8], [9]]
x2.shape = [4, 1]
且必须满足:len(x1.lod[0]) == len(x2.lod[0])
输出为 LoDTensor:
out.lod = [[0, 3+2, 5+4]]
out.data = [[1], [2], [3], [6], [7], [4], [5], [8], [9]]
out.shape = [9, 1]
参数¶
input (list of Variable) – 多个 LoDTensor 组成的 list,要求每个输入 LoDTensor 的 LoD 长度必须一致。数据类型为 float32、float64 或 int64。
name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor,输出连接后的 LoDTensor,数据类型和输入一致。
代码示例¶
import paddle
paddle.enable_static()
x = paddle.static.data(name='x', shape=[-1, 10], dtype='float32', lod_level=1)
y = paddle.static.data(name='y', shape=[-1, 10], dtype='float32', lod_level=1)
out = paddle.static.nn.sequence_concat(input=[x, y])