如何使用张量在TensorFlow中进行切片,如NumPy?

时间:2018-01-04 19:30:20

标签: python-3.x numpy tensorflow

我有两个张量,一个代表一个矢量列表,另一个代表一个与每个矢量对应的索引列表。我想获得相应索引处的每个向量的值。这就是我用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张量时,它不起作用。有没有办法得到相同的结果?

0 个答案:

没有答案