tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
更新时间：2020年01月04日 09:36:18 作者：u010551462
今天小编就为大家分享一篇tensorflow实现tensor中满足某一条件的数值取出组成新的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)
以上这篇tensorflow实现tensor中满足某一条件的数值取出组成新的tensor就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了pycharm远程linux开发和调试代码的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-07-07
Python判断远程服务器上Excel文件是否被人打开的方法
这篇文章主要介绍了Python如何判断远程服务器上Excel文件是否被人打开,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-07-07
这篇文章主要介绍了scrapy爬虫实例分享，分享了四则Python爬虫实例，具有一定借鉴价值,需要的朋友可以参考下2017-12-12
Python 一直以来被大家所诟病的一点就是执行速度慢，但不可否认的是 Python 依然是我们学习和工作中的一大利器。本文总结了15个tips有助于提升 Python 执行速度、优化性能，需要的可以参考一下2022-02-02
这篇文章主要介绍了简单了解Python生成器是什么，生成器就是一个在行为上和迭代器非常类似的对象，如果把迭代器比作 Android 系统，那么生成器就是 iOS，二者功能上差不多，但是生成器更优雅,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了利用Python探测附近WIFI密码,基于python脚本实现wifi密码的暴力破解从而实现免费蹭网，本文通过实例代码给大家介绍的非常详细，需要的朋友参考下吧2022-01-01
Python字符串格式化操作符(%)只适用于字符串类型，非常类似于C 语言里面的printf()函数的字符串格式化，甚至所用的符号都一样，都用百分号(%)，并且支持所有printf()式的格式化操作。2015-06-06
这篇文章主要介绍了python正则表达式re之compile函数解析，介绍了其定义，匹配模式等相关内容，具有一定参考价值，需要的朋友可以了解下。2017-10-10
本文主要介绍了Python和OpenCV库实现识别人物出现并锁定，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
今天小编就为大家分享一篇Python Numpy 自然数填充数组的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
最新评论