tf.where 和 tf.cond对比

tf.where 和 tf.cond都是类似于if…else 的逻辑操作，现对比两者区别

import tensorflow as tf 

pred=tf.placeholder(dtype=tf.bool,name='bool') 
x=tf.constant(1) 
y = tf.cond(pred,lambda:x+1,lambda:x-1) 
z=tf.where(pred,x+1,x-1) 
with tf.Session() as sess: 
	sess.run(tf.global_variables_initializer()) 
	y1,z1=sess.run([y,z],feed_dict={pred:True}) 
	y2,z2=sess.run([y,z],feed_dict={pred:False})

我们可以看出，用法几乎一样，不过，tf.cond针对的是函数，而tf.where针对的是具体计算

---

原作者：GAN_player
来源：优快云
原文：https://blog.youkuaiyun.com/GAN_player/article/details/77777644