tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
作者：u010551462 发布时间：2021-02-18 02:09:22
标签：tensorflow,tensor,数值
首先使用tf.where()将满足条件的数值索引取出来，在numpy中，可以直接用矩阵引用索引将满足条件的数值取出来，但是在tensorflow中这样是不行的。所幸，tensorflow提供了tf.gather()和tf.gather_nd()函数。
看下面这一段代码：
import tensorflow as tf
sess = tf.Session()
def get_tensor():
x = tf.random_uniform((5, 4))
ind = tf.where(x>0.5)
y = tf.gather_nd(x, ind)
return x, ind, y
在上述代码中，输出分别是原始的tensor x，x中满足特定条件(此处为>0.5)的数值的索引，以及x中满足特定条件的数值。执行以下步骤，观察三个tensor对应的数值：
x, ind, y = get_tensor()
x_, ind_, y_ = sess.run([x, ind, y])
可以得到如下结果：
可以看到，上述结果中将tensor x中大于0.5的数值取出来组成了一个新的tensor y。
如果我们将代码中的tf.gather_nd替换成tf.gather会发生什么呢？由于结果不方便展示，这里不放结果了，tf.gather适用于index为一维的情况，在本例中，index为2维，如果选用tf.gather的话，对应的x, ind, y的维数分别如下：
x.shape = (5, 4)
ind.shape = (9, 2)
y.shape = (9, 2, 4)
来源：https://blog.csdn.net/u010551462/article/details/90719545
0
投稿猜你喜欢
想实现发送邮件需要经过以下几步：1、登录邮件服务器2、构造符合邮件协议规则要求的邮件内容3、发送Python对SMTP支持有smtplib和- 1.字符串转二进制数组 string content="这是做个测试！"; System.Text.UnicodeEnc
- 本文实例讲述了Python实现调用另一个路径下py文件中的函数方法。分享给大家供大家参考，具体如下：针对这个问题，网上有很多的解决方式。其实
- 一、提出问题利用SQL，从右到左查找某一字符串中匹配的查询串的第一个索引位置。比如有一字段ProductName值格式如下：短袖印花T恤 M
- 一、问题描述当用JS调用form的方法submit直接提交form的时候，submit事件不响应。为什么？知道的请回复。类比一下，我用inp
- 在Python网络爬虫中，通常是通过TXT纯文本方式存储，其实也是可以存储在数据库中的；同时在WAMP（Windows、Apache、MyS
- 前言InnoDB采用按表空间（tablespace)的方式进行存储数据, 默认配置情况下会有一个初始大小为10MB， 名字为ibdata1的
- 由于某些原因需要把函数直接放到 img 标签上的 onload 属性执行，比如：For some reasons we have to ex
- 目录1. 文件相关函数2. 函数_函数的参数2.1 函数2.2 函数的参数3. 收集参数4. 命名关键字_总结小提示：5. 小练习练习问题：
- Vue路由跳转传参或打开新页面跳转1. 通过路由中的name属性 使用params传递参数, 使用this.$route.para
- 第一种方式阿里云、百度云服务器可用！！！yum install python3第二种方式1.下载python3.6.5的压缩包wget ht
- 这样做的好处是：利用表格来装载数据，不言而喻是最好的，你可以很灵活的为每个单元格定义样式。下面是具体的做法首先在photoshop设计一个效
- 特点：1.图片预载入，载入后再显示。意图一次呈现，不会让一块一块下载破坏你的页面，绝佳的用户体验，颠覆传统的浏览器呈现图片的处理方式（需要后
- python3.6下载地址： https://www.python.org/ftp/python/3.6.4/Python-3.6.4.tg
- 要使用多个定界符拆分字符串：使用 re.split() 方法，例如 re.split(r',|-', my_str)。re.
- jenkins是什么？ Jenkins是一个开
- replace 方法返回根据正则表达式进行文字替换后的字符串的复制。stringObj.replace(rgExp, replaceText
- Vue合并el-table第一列相同数据业务需求需要将el-table表格第一列相同的内容进行合并。解决办法el-table中使用 :spa
- 对于显存不充足的炼丹研究者来说，弄清楚Pytorch显存的分配机制是很有必要的。下面直接通过实验来推出Pytorch显存的分配过程。实验实验
- froglt 的站点：http://www.go2here.net 欢迎转载，请注明出处，未经作者允许，禁止一切商业应用。这是即
