tensorflow:将tensor中满足某一条件的数值取出组成新的tensor
首先使用tf.where()将满足条件的数值索引取出来,在numpy中,可以直接用矩阵引用索引将满足条件的数值取出来,但是在tensorflow中这样是不行的。所幸,tensorflow提供了tf.gather()和tf.gather_nd()函数。
看下面这一段代码:
import tensorflow as tf
sess = tf.Session()
def get_tensor():
...
原创
2019-05-31 17:27:59 ·
11012 阅读 ·
2 评论