codecamp

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具有相同的类型。

TensorFlow函数教程:tf.keras.backend.function
TensorFlow函数教程:tf.keras.backend.get_session
温馨提示
下载编程狮App,免费阅读超1000+编程语言教程
取消
确定