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就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python类的实例化问题解决,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
今天带大家了解python数据类型的相关知识,文中介绍的非常详细,对正在学习python的小伙伴们有很好地帮助,需要的朋友可以参考下2021-05-05
这篇文章主要给大家介绍了关于Python全栈之列表数据类型的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-10-10
这篇文章主要介绍了pytorch中使用LSTM，可以在troch.nn模块中找到LSTM类，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的朋友可以参考一下2022-07-07
在本篇内容里小编给大家整理的是一篇关于python中np的作用的相关文章，有兴趣的朋友们跟着学习下。2020-07-07
这篇文章主要介绍了Python 调用Java实例详解的相关资料,需要的朋友可以参考下2017-06-06
Python中的魔术方法Magic Methods使用实例全面指南
在Python中,魔术方法Magic Methods是一种特殊的方法,它们以双下划线开头和结尾,如__init__、__str__等,这些方法允许定制类的行为,使得对象更具有灵活性和可定制性,本文将深入探讨Python中一些常用的魔术方法,以及如何使用它们来定制类与对象2024-01-01
在Python中,字符串可以用单引号或双引号括起来,'hello' 与 "hello" 是相同的,这篇文章主要介绍了Python 中的字符串基础与应用,需要的朋友可以参考下2023-09-09
这篇文章主要介绍了linux之文件查找指定文件中包含关键字的行信息方式，具有很好的参考价值，希望对大家有所帮助。2023-06-06
这篇文章主要介绍了在django中实现choices字段获取对应字段值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
最新评论