有没有一种好的方法可以在张量流中进行“部分广播”?

时间:2018-06-26 18:43:54

标签: python tensorflow

如果我有以下计算顺序

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)

0 个答案:

没有答案