code: def tf_if(condition, a, b):
      
      int_condition = tf.to_float(tf.to_int64(condition))
      return a * int_condition + (1 - int_condition) * b
docs: Implements an if condition in tensorflow.
      :param condition: A boolean condition.
      :param a: Case a.
      :param b: Case b.
      :return: A if condition was true, b otherwise.