tf.gather_nd 实例

-Love-Coding-

于 2018-12-04 16:17:12 发布

阅读量597

点赞数 1

CC 4.0 BY-SA版权

分类专栏： TensorFlow

本文链接：https://blog.csdn.net/guotong1988/article/details/84790558

TensorFlow 专栏收录该内容

232 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文通过实例介绍了 TensorFlow 中的 tf.gather_nd 函数，展示了如何从多维张量中根据指定索引获取数据。示例中，我们看到不同形状的数据输出，包括2D和1D的结果，以及特定索引值的提取。

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

import tensorflow as tf

value = [[[1,2,3],[11,22,33]],[[4,5,6],[44,55,66]]]
init = tf.constant_initializer(value)
input = tf.get_variable('input', shape=[2,2,3], initializer=init)

value = [[[0,1],[1,0]]]
init = tf.constant_initializer(value)
index = tf.get_variable('index', shape=[2,1,2], initializer=init, dtype=tf.int32)

value = [[0,1],[1,0]]
init = tf.constant_initializer(value)
index2 = tf.get_variable('index2', shape=[2,2], initializer=init, dtype=tf.int32)

value = [0,1]
init = tf.constant_initializer(value)
index3 = tf.get_variable('index3', shape=[2], initializer=init, dtype=tf.int32)

value = [0,1,1]
init = tf.constant_initializer(value)
index4 = tf.get_variable('index4', shape=[3], initializer=init, dtype=tf.int32)

result = tf.gather_nd(input,index)
result2 = tf.gat