То, что вы предполагаете, неверно, оно имеет место, если условия и поток управления, это просто символ c. Например, tf.case :
f1 = lambda: tf.constant(17)
f2 = lambda: tf.constant(23)
r = tf.case([(tf.less(x, y), f1)], default=f2)
Его эквивалент этому коду:
if x < y:
return 17;
else:
return 23;
Существует также tf.switch_case для переключателя. заявления.