版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
tf.cond函数的使用
函数描述:
如果pred 为 true 则返回 true_fn() ,否则返回 false_fn()
cond (
pred ,
true_fn = None ,
false_fn = None ,
strict = False ,
name = None ,
fn1 = None ,
fn2 = None
)
示例:
z = tf.multiply( a , b )
result = tf.cond( X < Y, lambda:tf.add( X ,Z), lambda: tf.square( Y ))
如果X < Y,tf.add 将执行,并且 tf.square 操作不执行