如何从索引值中获取值

Question

张量为A_tensor，形状为[3,3,3]，我想通过索引获得最后一个轴的值。

如何在Tensorflow中做到这一点？

A_tensor =tf.constant([[1,2,3],[2,3,4],[3,4,5]])

如何得到张量([[1,2],[2,3],[3,4]])？

Answer 1

您可以使用tf.gather():

A_tensor =tf.constant([[1,2,3],[2,3,4],[3,4,5]])
tf.gather(A_tensor,[[0,1]],axis=-1)