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

由 Carrie 创建, 最后一次修改 2019-03-13

tf.keras.backend.gather函数

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

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

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

参数:

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

返回:

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

以上内容是否对您有帮助:

二维码
建议反馈
二维码