В numpy мы можем выполнять многоиндексирование, например
sample = np.zeros([3,10,10,100])
idx = [10, 80, 20, 12, 9]
sample_idx = sample[:,:,:,idx]
print(sample_idx.shape)
, и получаем (3, 10, 10, 5) здесь.
Но в тензорном потоке это не такне работает с таким же образом, даже с tf.gather_nd .
Итак, есть ли способ сделать это в тензорном потоке?