Tensorflow-API ： tf.gather

Elag

于 2018-05-23 16:05:26 发布

阅读量410

点赞数 1

分类专栏： Tensorflow-API python

本文链接：https://blog.youkuaiyun.com/u011436429/article/details/80421415

版权

python 同时被 2 个专栏收录

21 篇文章

订阅专栏

Tensorflow-API

5 篇文章

订阅专栏

本文介绍TensorFlow中的tf.gather函数，该函数可以根据指定索引从输入张量的特定轴上收集切片。通过一个具体例子展示了如何使用tf.gather从三维张量中选取特定索引的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.gather

根据索引从参数轴上收集切片,索引必须是任何维度的整数张量 (通常为 0-D 或 1-D)

import tensorflow as tf

t1 = tf.reshape(tf.range(0,16),[2,2,4])
# [[[ 0  1  2  3]
#   [ 4  5  6  7]]
#
#  [[ 8  9 10 11]
#   [12 13 14 15]]]

# 取第2维中，index为1,3的数据
t2 = tf.gather(t1,[1,3],axis=2)

with tf.Session() as sess:
    print(sess.run(t1))
    print(sess.run(t2))

[[[ 1  3]
  [ 5  7]]

 [[ 9 11]
  [13 15]]]