首页 > 其他分享 >tf.gather,取指定维度多个索引的数据

tf.gather,取指定维度多个索引的数据

时间:2022-10-31 16:34:37浏览次数:47  
标签:tensor 索引 gather 维度 tf array


tensorflow和numpy在数据处理上语法相似但又不完全一样,比如在numpy中想取指定维度的多个指定索引所指向的数据时,直接用一个列表保存索引就能直接取,比如:

# b的shape为[2, 3, 2]
b = np.array([[[1, 2], [2, 3], [3,4]], [[4, 5], [5,6], [6, 7]]])
a = [0, 1, 2]
# 假若要取b的第2个维度(从1算起)的以a为索引的数据,只需写成如下形式
b[:, a, :]

但是若b是tensor形式,则上述操作会报错!

TypeError: Only integers, slices (`:`), ellipsis (`...`), tf.newaxis (`None`) and scalar tf.int32/tf.int64 tensors are valid indices, got array([0, 1, 2])

而在tensor中则需要使用tf.gather方法来实现该功能。
tf.gather可以取指定维度任意索引的数据,举个例子:

b = np.array([[[1, 2], [2, 3], [3,4]], [[4, 5], [5,6], [6, 7]]])
b = tf.constant(b)
a = [0, 1, 2]
# 写法如下,效果和上面的numpy的效果相同
tf.gather(b, axis=1, indices=a)

这是tf.gather的用法,而想来大家也都听过tf.gather_nd,这个功能更加强大,它能对同一个维度的每一组数据分别取不同的索引值,比如取第1行的第二列+第二行的第三列。
关于gather_nd的用法见:​​tensorflow中tensor,从每行取指定索引元素​​


标签:tensor,索引,gather,维度,tf,array
From: https://blog.51cto.com/u_15854687/5810311

相关文章

  • [BUUCTF]第八天训练日志
    文章目录​​[RoarCTF2019]EasyJava​​​​[BUUCTF2018]OnlineTool​​​​[GYCTF2020]Blacklist​​​​[MRCTF2020]你传你......
  • [BUUCTF]第八天训练日记
    文章目录​​[ZJCTF2019]NiZhuanSiWei​​​​[CISCN2019华北赛区Day2Web1]HackWorld​​​​[网鼎杯2018]Fakebook​​​​[极客大挑战2019]HardSQL​​​​[GXYCTF......
  • [BUUCTF]第六天训练日志
    文章目录​​[SUCTF2019]EasyWeb​​​​[HFCTF2020]EasyLogin​​​​[CISCN2019华北赛区Day2Web1]HackWorld​​[SUCTF2019]EasyWeb做了一半做不动了,换个时间来,花费......
  • [BUUCTF]第七天训练日志
    [SUCTF2019]EasyWeb​​suctf2019​​​​bypassopen_basedir的新方法​​阿姨熄灯了睡觉......
  • [BUUCTF]第十二天训练日志
    文章目录​​[SWPU2019]Web1​​[SWPU2019]Web1首先注册一个​​admin​​​用户提示已经被注册,估计要伪造admin身份了,发布广告那一页​​<script>alert(1);</scirpt>​​......
  • [BUUCTF]第三天训练日志
    文章目录​​[GKCTF2020]EZ三剑客-EzWeb​​​​[FBCTF2019]RCEService​​​​[BSidesCF2019]Kookie​​​​[CISCN2019华北赛区Day1Web5]CyberPunk​​​​[极客大挑......
  • [BUUCTF]第一天训练
    文章目录​​[BSidesCF2019]Futurella​​​​[BJDCTF2nd]duangShell​​​​反弹姿势1(未成功)​​​​反弹姿势2​​[BSidesCF2019]Futurella好久没看过这么简单的题目......
  • day-05 接口架构风格 —RESTful
    第五章接口架构风格—RESTful认识RESTREST(英文:RepresentationalStateTransfer,简称REST)一种互联网软件架构设计的风格,但它并不是标准,它只是提出了一组客户端和服......
  • RobotFramework基本语法----list
    1.遍历${board_slotid}   createlistFOR   ${slot}   INRANGE  20  AppendtoList  ${board_slotid}  ${slot}ENDFOR  ${i}......
  • 使用Three.js 心得系列一 如何创建三维网格三、使用Three.js 心得系列三 Three.js 如
    一、使用Three.js心得系列一如何创建三维网格二、使用Three.js心得系列二如何改变三维场景的GlTF模型的位置三、使用Three.js心得系列三Three.js如何加载GlTF文件 ......