tensorflow实现tensor中满足某一条件的数值取出组成新的tensor


Posted in Python onJanuary 04, 2020

首先使用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])

可以得到如下结果:

tensorflow实现tensor中满足某一条件的数值取出组成新的tensor

tensorflow实现tensor中满足某一条件的数值取出组成新的tensor

可以看到,上述结果中将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)

以上这篇tensorflow实现tensor中满足某一条件的数值取出组成新的tensor就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python中marshal对象序列化的相关知识
Jul 01 Python
Djang中静态文件配置方法
Jul 30 Python
基于Python实现通过微信搜索功能查看谁把你删除了
Jan 27 Python
深入理解Python中的内置常量
May 20 Python
分享一个可以生成各种进制格式IP的小工具实例代码
Jul 28 Python
django 常用orm操作详解
Sep 13 Python
python PIL和CV对 图片的读取,显示,裁剪,保存实现方法
Aug 07 Python
使用Python的datetime库处理时间(RPA流程)
Nov 24 Python
详解从Django Allauth中进行登录改造小结
Dec 18 Python
Python 余弦相似度与皮尔逊相关系数 计算实例
Dec 23 Python
Python urllib库如何添加headers过程解析
Oct 05 Python
Python制作运行进度条的实现效果(代码运行不无聊)
Feb 24 Python
对tensorflow中的strides参数使用详解
Jan 04 #Python
tensorflow之获取tensor的shape作为max_pool的ksize实例
Jan 04 #Python
TensorFlow tf.nn.max_pool实现池化操作方式
Jan 04 #Python
TensorFlow tf.nn.conv2d实现卷积的方式
Jan 03 #Python
Python调用钉钉自定义机器人的实现
Jan 03 #Python
pytorch中的上采样以及各种反操作,求逆操作详解
Jan 03 #Python
pytorch 获取tensor维度信息示例
Jan 03 #Python
You might like
PHP5中新增stdClass 内部保留类
2011/06/13 PHP
深入解读php中关于抽象(abstract)类和抽象方法的问题分析
2014/01/03 PHP
PHP empty函数报错解决办法
2014/03/06 PHP
WordPress主题制作中自定义头部的相关PHP函数解析
2016/01/08 PHP
浅谈PHP发送HTTP请求的几种方式
2017/07/25 PHP
PHP简单实现二维数组赋值与遍历功能示例
2017/10/19 PHP
PHP+Session防止表单重复提交的解决方法
2018/04/09 PHP
PHP实现负载均衡的加权轮询方法分析
2018/08/22 PHP
从零开始学习jQuery (二) 万能的选择器
2010/10/01 Javascript
js实现仿微博滚动显示信息的效果
2015/12/21 Javascript
angularjs表格分页功能详解
2016/01/21 Javascript
BootStrap 智能表单实战系列(五) 表单依赖插件处理
2016/06/13 Javascript
jQuery基于ajax方式实现用户名存在性检查功能示例
2017/02/10 Javascript
JavaScript门道之标准库
2018/05/26 Javascript
微信小程序实现漂亮的弹窗效果
2020/05/26 Javascript
echarts大屏字体自适应的方法步骤
2019/07/12 Javascript
wxPython事件驱动实例详解
2014/09/28 Python
Pycharm远程调试openstack的方法
2017/11/21 Python
pycharm远程调试openstack代码
2017/11/21 Python
Python操作mongodb的9个步骤
2018/06/04 Python
python用BeautifulSoup库简单爬虫实例分析
2018/07/30 Python
python框架中flask知识点总结
2018/08/17 Python
Python多进程入门、分布式进程数据共享实例详解
2019/06/03 Python
python线程安全及多进程多线程实现方法详解
2019/09/27 Python
将matplotlib绘图嵌入pyqt的方法示例
2020/01/08 Python
Django实现前台上传并显示图片功能
2020/05/29 Python
服装行业创业计划书范文
2014/02/05 职场文书
成绩单公证书
2014/04/10 职场文书
反对四风问题自我剖析材料
2014/09/29 职场文书
民主评议政风行风活动心得体会
2014/10/29 职场文书
城南旧事观后感
2015/06/11 职场文书
无犯罪记录证明样本
2015/06/16 职场文书
2015年机关作风和效能建设工作总结
2015/07/23 职场文书
解决jupyter notebook图片显示模糊和保存清晰图片的操作
2021/04/24 Python
Python Django搭建文件下载服务器的实现
2021/05/10 Python
Vue OpenLayer 为地图绘制风场效果
2022/04/24 Vue.js