map_structure使用问题
Created by: Angus07
我有两个tensor,分别是 index: [batch_size ,1600, 1] update: [batch_size, 1600] 希望能通过scatter_nd变成: output: [batch_size, 18000] 在tensorflow中可以通过tf.map_fn实现。现在使用paddle中的utils.map_structure(没有相关文档)来复现, attn_dists_projected = utils.map_structure(lambda y: layers.scatter_nd(y[0], y[1], [18000]), (index, update)) 结果报错了,报错信息如下:
希望能告知正确的用法,或者其他可以实现这个功能的办法。