TensorFlow函數(shù)教程：tf.keras.backend.gather

tf.keras.backend.gather函數(shù)

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

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

檢索張量reference中索引indices的元素。

參數(shù)：

• reference：一個張量。
• indices：索引的整數(shù)張量。

返回：

返回一個張量，它與reference具有相同的類型。