深度理解tf.gather和tf.gather_nd的用法

最新推荐文章于 2022-12-07 23:23:36 发布
最新推荐文章于 2022-12-07 23:23:36 发布
阅读量1.3w 收藏 13
点赞数 4
分类专栏： python开发 tensorflow NLP
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/oksupersonic/article/details/104559821
版权
本文对比了tf.gather和tf.gather_nd在TensorFlow中的使用，阐述了两者从tensor中提取指定index部分的功能。gather适用于单一index标注，而gather_nd支持多级index标注，能更灵活地处理复杂数据结构。

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

tf.gather和tf.gather_nd都是从tensor中取出index标注的部分，不同之处在于，gather一般只使用一个index来标注，而gather_nd可以使用多个index。
例如：
在这里插入图片描述
在这里插入图片描述
如何直观理解gather_nd的indices呢？
在上例中，直观的理解就是，取出params中位于[0,0]和[1,1]处的tensor，放入index中对应的位置。
换句话说，除去tensor维之外，返回值的形状和indices相同，值由indices标注。
如果理解了这一点，就可以用gather_nd实现gather：
在这里插入图片描述

tf.gather_nd 实例
_Hope_
12-04 578
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...
tf.gather_nd和tf.gather的区别与联系
blingkeyholic的博客
12-16 2863
tf.gather_nd和tf.gather的区别与联系 tf.gather_nd和tf.gather都是选择其中需要的部分组成一个新的矩阵。 其中tf.gather(params,indices,axis=0.name=None)是选择相同维度的元素组成新的矩阵，params是待处理tensor,indices是一个1-D tensor表示对应轴上的索引，axis是指定维度，name是操作名字。...
Tensorflow常用函数
jiangjiane
07-21 537
tf.concat 把一组向量从某一维上拼接起来，很向numpy中的Concatenate： t1 = [[1, 2, 3], [4, 5, 6]] t2 = [[7, 8, 9], [10, 11, 12]] tf.concat([t1, t2], 0) ==> [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]] tf.concat([t1...
tf.gather_nd函数
weixin_49346755的博客
07-12 2742
函数说明 从指定的张量中根据索引来进行切片。注意tf.gather函数和tf.gather_nd函数的区别。参数params表示待切片的张量。参数indices表示进行切片的索引。参数batch_dims表示批处理的维度，这个参数比较难理解，可以看一下使用示例中的用法。1、一维张量的索引 2、二维张量的索引 3、二维张量的切片 4、三维张量的索引 5、三维张量的切片...
tf.gather和tf.gather_nd
qq_40657582的博客
11-23 918
tf.gather Gather slices from params axis according to indices. tf.gather(params, indices, validate_indices=None, axis=None, batch_dims=0, name=None) Example： # data: [classes, students, subjects] data = tf.ones([4,35,8]) print(data.shape) # TensorShap
tf.gather_nd 用法
热门推荐
崔伟灿的博客
12-11 1万+
tf.gather_nd的函数原型是： def gather_nd(params, indices, name=None) 根据定义， 其主要功能是根据indices描述的索引，提取params上的元素， 重新构建一个tensor 在谈论该函数之前，我们先来看一下 索引的概念， 在一维数组中，元素的索引即该元素在数组中序号，通常序号从0开始标记 如数组 ary=[1,2,3,4]; ...
Pytorch 实现tf.gather()
rosefun96的博客
01-02 2651
1. 实现tf.gather 在pytorch中，实现 tf.gather 很简单，只需要使用 select。 select(dim, index) → Tensor 比如， import numpy as np a = np.array([[1],[2],[3],[4],[5]]) b = torch.from_numpy(a) indices = [ 1, 2, 0] b[indices] ...
tf.gather_nd怎么用pytorch实现相同功能
最新发布
10-21
在PyTorch中，如果你想实现类似于TensorFlow `tf.gather_nd`的功能，可以使用`torch.index_select`或者`torch.scatter`结合`torch.nonzero`函数。`tf.gather_nd`用于从张量中选择指定索引的元素，而`index_select`则...
Tensorflow （一）： scatter_nd 与 gather_nd 的应用
weixin_43509924的博客
03-01 535
Tensorflow scatter_nd 与 gather_ndTensorflow scatter_nd 与 gather_nd Tensorflow scatter_nd 与 gather_nd 图片HxW 中 每行中挑选 top k 大的像素， 其余的像素设为0 选取top_k x = tf.constant([[6., 2., 0.], [0., 4., 5.]]) # of type tf.float32 k = 2 values, indices = tf.math.top_k(x, k,
tf.gather_nd用法详解
tuying_001的专栏
10-21 2307
gather_nd的定义如下： def gather_nd(params, indices, name=None) 功能：根据indeces描述的索引，在params中提取元素，重新组成一个tansor ``
tf.gather, tf.gather_nd和tf.slice
机器学习杂货铺1号店
12-29 3194
在tensorflow和pytorch中，都有一些函数是用于对张量进行索引的，因为pytorch是基于动态图的，它的张量可以动态的进行操作，不需要在对话(session)中进行，因此可能采用这些方法的必要性没有那么足，但是因为tensorflow是基于静态图的，因此如果在构建计算图的过程中需要对张量进行索引，一般是不能进行简单的slice的，因此引入这些函数就很有必要了，我们这里集中介绍下这些函数...
【Tensorflow】gather与gather_nd
littlemichelle
12-07 818
tf1.x 类比切片目录tf.one_hot 提取行gather 提取行 gather 提取列gather_nd与lookup的作用差不多embedding层_tensorflow中的Embedding操作详解_weixin_39835321的博客-优快云博客 运行时，上面的遍历才会产生 gather 提取行 可以看到gather也实现了如上tf.one_hot的效果 gather 提取列 当emb是一维时 gather_nd tf.gather函数呢，我们只能通过
tf.gather和tf.gather_nd的使用和区别
醉意流年go的博客
06-17 497
# coding:utf-8 """ @author: liu @File: tf_gather_nd.py @CreateTime: 2021/6/17 """ """ 测试tf.gather和tf.gather_nd的区别和联系 """ import numpy as np import tensorflow as tf print("tensorflow version: ", tf.__version__) data = np.arange(18).reshape((3, 2, 3).
tf.gather tf.gather_nd 和 tf.batch_gather 使用方法
张冰洋的天空
01-19 1万+
    在计算机视觉的项目中，会遇到根据索引获取数组特定下标元素的任务。常用的函数有tf.gather , tf.gather_nd 和 tf.batch_gather。 1.tf.gather(params,indices,validate_indices=None,name=None,axis=0) 主要参数：params：被索引的张量                     indic...
tf.gather_nd()
dili8870的博客
07-31 347
tf.gather_nd( 　　params, 　　indices, 　　name=None, 　　batch_dims=0) TensorFlow链接：https://tensorflow.google.cn/api_docs/python/tf/gather_nd?hl=en 功能：将参数中的切片收集到由索引指定的形状的张量中. 参数： params：张...
tf.gather_nd
weixin_36670529的博客
09-02 1517
将params中的切片收集到一个由指标指定形状的张量中。 tf.gather_nd( params, indices, batch_dims=0, name=None ) 指标是一个k维整数张量，最好考虑为(K-1)张量的指标到帕拉姆，其中每个元素定义了帕拉姆的一个切片: output[\\(i_0, ..., i_{K-2}\\)] = params[...
【TensorFlow】tf.gather_nd
mjiansun的专栏
10-23 894
函数：tf.gather_nd gather_nd( params, indices, name=None ) 参见指南：张量变换>分割和连接 将参数中的切片收集到由索引指定的形状的张量中. 索引(indices)是一个 k 维整数张量,最好作为一个 (k-1) 维的索引(indices)张量的参数,其中每个元素定义了一个参数切片： output[i_...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值