def drop_path(columns, coin): with tf.variable_op_scope([columns], None, "DropPath"): out = tf.cond(coin, lambda : drop_some(columns), lambda : random_column(columns)) return out