sequence_scatter¶

paddle.static.nn. sequence_scatter ( input, index, updates, name=None ) ¶

根据 index 提供的位置将 updates 中的信息更新到输出中。

先使用 input 初始化 output，然后通过 output[instance_index][index[pos]] += updates[pos]方式，将 updates 的信息更新到 output 中，其中 instance_idx 是 pos 对应的在 batch 中第 k 个样本。

output[i][j]的值取决于能否在 index 中第 i+1 个区间中找到对应的数据 j，若能找到 out[i][j] = input[i][j] + update[m][n]，否则 out[i][j] = input[i][j]。

例如，在下面样例中，index 的 lod 信息分为了 3 个区间。其中，out[0][0]能在 index 中第 1 个区间中找到对应数据 0，所以，使用 updates 对应位置的值进行更新，out[0][0] = input[0][0]+updates[0][0]。out[2][1]不能在 index 中第 3 个区间找到对应数据 1，所以，它等于输入对应位置的值，out[2][1] = input[2][1]。

样例:

输入：

input.data = [[1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
              [1.0, 1.0, 1.0, 1.0, 1.0, 1.0],
              [1.0, 1.0, 1.0, 1.0, 1.0, 1.0]]
input.dims = [3, 6]

index.data = [[0], [1], [2], [5], [4], [3], [2], [1], [3], [2], [5], [4]]
index.lod =  [[0,        3,                       8,                 12]]

updates.data = [[0.3], [0.3], [0.4], [0.1], [0.2], [0.3], [0.4], [0.0], [0.2], [0.3], [0.1], [0.4]]
updates.lod =  [[  0,            3,                                 8,                         12]]

输出：

out.data = [[1.3, 1.3, 1.4, 1.0, 1.0, 1.0],
            [1.0, 1.0, 1.4, 1.3, 1.2, 1.1],
            [1.0, 1.0, 1.3, 1.2, 1.4, 1.1]]
out.dims = X.dims = [3, 6]