keras--backend.gather

tf.keras.backend.gather(reference, indices) 函数用于检索张量reference中索引indices的元素。

import tensorflow as tf
indices = [0, 2]
reference = [1,2,3,4]
tf.keras.backend.gather(reference, indices)
>>> array([1, 3])