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用10行代码实现对黄色图片的检测功能
Aug 10 Python
linux环境下的python安装过程图解(含setuptools)
Nov 22 Python
python学生信息管理系统(初级版)
Oct 17 Python
Python实现带下标索引的遍历操作示例
May 30 Python
pyqt5中QThread在使用时出现重复emit的实例
Jun 21 Python
详解python和matlab的优势与区别
Jun 28 Python
python实现数据分析与建模
Jul 11 Python
Python pip 安装与使用(安装、更新、删除)
Oct 06 Python
python列表生成器迭代器实例解析
Dec 19 Python
tensorflow的ckpt及pb模型持久化方式及转化详解
Feb 12 Python
Python如何脚本过滤文件中的注释
May 27 Python
Python paramiko使用方法代码汇总
Nov 20 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
ADODB类使用
2006/11/25 PHP
php HandlerSocket的使用
2011/05/02 PHP
深入探讨:PHP使用数据库永久连接方式操作MySQL的是与非
2013/06/05 PHP
使用php清除bom示例
2014/03/03 PHP
PHP中is_dir()函数使用指南
2015/05/08 PHP
PHP crc32()函数讲解
2019/02/14 PHP
PHP 模拟登陆功能实例详解
2019/09/10 PHP
一个很简单的办法实现TD的加亮效果.
2006/06/29 Javascript
JavaScript 输入框内容格式验证代码
2010/02/11 Javascript
PHP+jQuery+Ajax+Mysql如何实现发表心情功能
2015/08/06 Javascript
AngularJS自定义插件实现网站用户引导功能示例
2016/11/07 Javascript
windows下vue-cli及webpack搭建安装环境
2017/04/25 Javascript
微信小程序中input标签详解及简单实例
2017/05/18 Javascript
axios拦截设置和错误处理方法
2018/03/05 Javascript
JQuery模拟实现网页中自定义鼠标右键菜单功能
2018/11/14 jQuery
python制作一个桌面便签软件
2015/08/09 Python
python爬取亚马逊书籍信息代码分享
2017/12/09 Python
浅谈Python使用Bottle来提供一个简单的web服务
2017/12/27 Python
python实现媒体播放器功能
2018/02/11 Python
python3中的md5加密实例
2018/05/29 Python
python实现飞机大战
2018/09/11 Python
Python中的asyncio代码详解
2019/06/10 Python
Python generator生成器和yield表达式详解
2019/08/08 Python
Python控制台输出时刷新当前行内容而不是输出新行的实现
2020/02/21 Python
使用卷积神经网络(CNN)做人脸识别的示例代码
2020/03/27 Python
django 模型中的计算字段实例
2020/05/19 Python
Python logging模块写入中文出现乱码
2020/05/21 Python
Timberland法国官网:购买靴子、鞋子、衣服、夹克和配饰
2019/11/30 全球购物
在weblogic中发布ejb需涉及到哪些配置文件
2012/01/17 面试题
考试违纪检讨书
2014/02/02 职场文书
艺术节主持词
2014/04/02 职场文书
2014年专项整治工作总结
2014/11/17 职场文书
项目合作意向书
2015/05/08 职场文书
导游词之五台山
2019/10/11 职场文书
GPU服务器的多用户配置方法
2022/07/07 Servers
SQL Server数据库的三种创建方法汇总
2023/05/08 MySQL