Conditional statements in TF; where()

Introduction

tf.where(conditional statement, A, B)

Return A if the conditional statement is true, return B if it is false

Code

import tensorflow as tfa = tf.constant([1, 2, 3, 4, 99, 6])b = tf.constant([2, 3, 4, 5, 6, 7])c = tf.where(tf.greater(a, b), a, b)print(c)