tensorflow的用法：tf.where 和 tf.cond对比

import tensorflow as tf
p=tf.constanc(4)
x=tf.constant(5)
y = tf.cond(p<x,lambda:x**2,lambda:x)
z=tf.where(p<x,x**2,x)

with tf.Session() as sess:

    sess.run(tf.global_variables_initializer())
    y1,z1=sess.run([y,z])
    y2,z2=sess.run([y,z])

可以看出，tf.where和tf.cond可以实现一样的功能，只是tf.cond两个值必须要是函数，tf.where内部的两个值必须要是tensor值。
参考https://blog.csdn.net/GAN_player/article/details/77777644