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

2019-03-13 17:33 更新

tf.keras.backend.gather函数

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

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

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

参数:

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

返回:

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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号