TensorFlow函数教程：tf.keras.backend.gather

tf.keras.backend.gather函数

tf.keras.backend.gather(
    reference,
    indices
)

定义在：tensorflow/python/keras/backend.py。

检索张量reference中索引indices的元素。

参数：

• reference：一个张量。
• indices：索引的整数张量。

返回：

返回一个张量，它与reference具有相同的类型。