使用像 numpy 这样的索引从张量中提取

Question

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

Answer 1

您可以通过以下方式完成tf.gather：

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