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正则替换字符串函数re.sub用法示例
Jan 19 Python
python实现将excel文件转化成CSV格式
Mar 22 Python
解决python写入带有中文的字符到文件错误的问题
Jan 31 Python
使用python接入微信聊天机器人
Mar 31 Python
使用Python脚本zabbix自定义key监控oracle连接状态
Aug 28 Python
用Python写一个自动木马程序
Sep 17 Python
pyspark给dataframe增加新的一列的实现示例
Apr 24 Python
Python预测2020高考分数和录取情况
Jul 08 Python
python从ftp获取文件并下载到本地
Dec 05 Python
python中print格式化输出的问题
Apr 16 Python
python获取带有返回值的多线程
May 02 Python
Python使用Beautiful Soup(BS4)库解析HTML和XML
Jun 05 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
php 表单数据的获取代码
2009/03/10 PHP
php模拟服务器实现autoindex效果的方法
2015/03/10 PHP
用js模拟JQuery的show与hide动画函数代码
2010/09/20 Javascript
关于onScroll事件在IE6下每次滚动触发三次bug说明
2011/09/21 Javascript
jquery 无限级联菜单案例分享
2013/03/26 Javascript
JS获取地址栏参数的小例子
2013/08/23 Javascript
jQuery DOM操作实例
2014/03/05 Javascript
jQuery中的编程范式详解
2014/12/15 Javascript
jQuery处理图片加载失败的常用方法
2015/06/08 Javascript
jquery仿百度百科底部浮动导航特效
2015/08/08 Javascript
js去字符串前后空格的实现方法
2016/02/26 Javascript
JavaScript实现弹出DIV层同时页面背景渐变成半透明效果
2016/03/25 Javascript
Bootstrap打造一个左侧折叠菜单的系统模板(二)
2016/05/17 Javascript
Bootstrap组件系列之福利篇几款好用的组件(推荐)
2016/06/23 Javascript
AngularJS框架的ng-app指令与自动加载实现方法分析
2017/01/04 Javascript
Javascript 严格模式use strict详解
2017/09/16 Javascript
JavaScript实现异步图像上传功能
2018/07/12 Javascript
vue 自定指令生成uuid滚动监听达到tab表格吸顶效果的代码
2020/09/16 Javascript
[00:12]DAC2018 天才少年转战三号位,他的SOLO是否仍如昔日般强大?
2018/04/06 DOTA
Python读写配置文件的方法
2015/06/03 Python
Django读取Mysql数据并显示在前端的实例
2018/05/27 Python
Python正则表达式实现简易计算器功能示例
2019/05/07 Python
Python实现Mysql数据统计及numpy统计函数
2019/07/15 Python
详解Python文件修改的两种方式
2019/08/22 Python
CSS3教程(4):网页边框和网页文字阴影
2009/04/02 HTML / CSS
Html5获取高德地图定位天气的方法
2019/12/26 HTML / CSS
Skyscanner澳大利亚:全球领先的旅游搜索网站
2018/03/24 全球购物
切尔西足球俱乐部官方网上商店:Chelsea FC
2019/06/17 全球购物
毕业证代领委托书
2014/09/26 职场文书
群众路线四风对照检查材料
2014/11/04 职场文书
应届生求职自荐信范文
2015/03/04 职场文书
工作简报格式范文
2015/07/21 职场文书
python使用tkinter实现透明窗体上绘制随机出现的小球(实例代码)
2021/05/17 Python
详解python网络进程
2021/06/15 Python
python unittest单元测试的步骤分析
2021/08/02 Python
html网页引入svg图片的4种方式
2022/08/05 HTML / CSS