根據Tensorflow中的其他特定張量重新排列張量

Question

給定某些張量a和b ，其中b的類型為uint且max(b) < len(a) ，我想得到張量c ：

c = a[b]

這實際上a根據b給出的順序重新排列張量b 。

Answer 1

只是為了結束問題-根據@jdehesa的評論：

c = tf.gather(a, b)

這正是需要的。 謝謝@jdehesa！