TF中的索引操作

Question

有没有办法在张量流中索引操作？特别是，我对通过tf.while_loop的迭代器变量建立索引感兴趣。

更具体地说，假设我有my_ops = [op1, op2]。我想拥有：

my_ops = [...]
i = tf.constant(0)
c = lambda i: tf.less(i, 10)
b = lambda i: my_ops[i](...)
r = tf.while_loop(c, b, [i])

不幸的是，这无法正常工作，因为python数组仅支持整数索引。

Answer 1

我相信这是不可能的。但是，您可以改为使用tf.stack来堆叠操作的输出张量，然后使用tf.gather获得所需的输出。

这里有个例子：

main