目录

tf.greater(v1,v2)

tf.where(p1,p2,p3)


tf.greater(v1,v2)

功能:比较两个输入张量的每一个元素的大小,并返回比较结果。


import tensorflow as tf

v1 = tf.constant([1,2,3,4])
v2 = tf.constant([4,3,2,1])

with tf.Session() as sess:
    print(sess.run(tf.greater(v1,v2)))   
    sess.close()

输出结果:[False False  True  True]


 


tf.where(p1,p2,p3)

参数p1是一个bool变量,也可以是一个表达式,返回值是True或者False。这个变量可以是一个也可以是一个列表,就是很多个True或者False组成的列表。

如果是True,返回p2,反之返回p3


#coding:utf-8

import tensorflow as tf

A = 100
B = tf.constant([1, 2, 3, 4])
C = tf.constant([1, 1, 1, 1])
D = tf.constant([0, 0, 0, 0])

with tf.Session() as sess:
    print(sess.run(tf.where(A > 1, 'A', 'B')))
    print(sess.run(tf.where(False, 'A', 'B')))
    print(sess.run(tf.where(B > 2, C, D)))

 

输出:b'A'
            b'B'
            [0 0 1 1]