Question

我有一个张量，例如形状为（1,20,4）的张量1。我正在尝试使用此张量的某些索引（1、4、5）创建一个张量。我可以使用tensor [：，[1,4,5] ,:]来做这种形式的numpy。据我了解，可以使用“ tf.gather_nd”完成此操作，但我真的不知道该如何完成。

Answer 1

您可以使用tf.gather完成以下操作：

tensor2 = tf.gather(tensor1, [1, 4, 5], axis=1)