Python 利用numpy等指标提取张量

我有一个张量，比如形状为（1,20,4）的张量1。我试图用这个张量的某些指数（1，4，5）来创建一个张量。例如，我可以用张量[：，[1,4,5]，：]来表示numpy。据我所知，这可以使用“tf.gather\n d”来完成，但我真的不知道如何完成。

---

我想你要找的是

tf.slice（）

。这可以用来得到张量的某个元素@明白了你有什么问题吗？如果这个解决方案在你的案例中不起作用，也许你可以发布一个新问题。

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