sequence_enumerate¶
枚举形状为 [d_1, 1]
的输入序列所有长度为 win_size
的子序列,生成一个形状为 [d_1, win_size]
的新序列,需要时以 pad_value
填充。
该 API 的输入 input
只能是 LodTensor。
范例如下:
给定输入 x:
x.lod = [[0, 3, 5]]
x.data = [[1], [2], [3], [4], [5]]
x.dims = [5, 1]
设置属性 win_size = 2 pad_value = 0
得到输出 out:
out.lod = [[0, 3, 5]]
out.data = [[1, 2], [2, 3], [3, 0], [4, 5], [5, 0]]
out.dims = [5, 2]
参数¶
input (Variable)- 输入序列,形状为
[d_1, 1]
,lod level 为 1 的 LodTensor。数据类型支持 int32,int64,float32 或 float64。win_size (int)- 子序列窗口大小。
pad_value (int,可选)- 填充值,默认为 0。
name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
枚举序列,形状为 [d_1, win_size]
,lod_level 为 1 的 LoDTensor。数据类型与输入 input
一致。
代码示例¶
import paddle
paddle.enable_static()
x = paddle.static.data(name='x', shape=[-1, 1], dtype='int32', lod_level=1)
out = paddle.static.nn.sequence_enumerate(input=x, win_size=3, pad_value=0)