我有两个张量,一个代表一个矢量列表,另一个代表一个与每个矢量对应的索引列表。我想获得相应索引处的每个向量的值。这就是我用numpy做的事情:
vectors = np.array([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14]])
indices = np.array([1, 4, 3])
values = vectors[np.arange(3), indices]
# values: [1, 9, 13]
但是当我尝试“矢量”和“索引”是TensorFlow张量时,它不起作用。有没有办法得到相同的结果?