如果我有以下计算顺序
origvals = [3, 9, 2, 8, 5, 9, 1, 6, 3, 2]
updates = [2, 5, 9, 8]
indices = [0, 0, 0, 0, 1, 1, 1, 1, 1, 2]
updates_pb = [2, 2, 2, 2, 5, 5, 5, 5, 5, 9] # tf.gather(updates, indices)
result = origvals + updates_pb
在这里,我根据更新值和索引显示了updates_pb,它是对值“ 2”,“ 5”和“ 9”的分段广播。在我的用例中,几乎总是在索引向量中有长条重复值。
有没有比通过高度重复的update_pb张量的中间构造更有效的方法来计算最终结果值?愿望清单功能可能是:
result = tf.gather_add(origvals, updates, indices)