tensorflow实现if/else
#如果对数里面的值<0,将其置1，对数项为0t1 = tf.cond(tf.less(x, 0), lambda: tf.assign(x, 1.), lambda: x)
import tensorflow as tf
假设t1是一个Tensor
t1 = …
定义条件函数
def condition():
# 当t1<0时执行的操作
return tf.constant(-1)
def body():
# 当t1>=0时执行的操作
return tf.constant(1)
使用tf.cond判断条件并执行操作
result = tf.cond(tf.less(t1, 0), condition, body)
