qvdgeer

trouble using tf.where

3

Hi, trying to get a toy example with tf.where working using TensorFlow sess = TensorFlow.Session() A = TensorFlow.Variable([-1,-1,-1,-1,0,-1]) B = TensorFlow.Variable([0,0,0,0,1,0]) mask = TensorFlow.equal(A,0) C= TensorFlow.where(mask,B,A) run(sess, TensorFlow.global_variables_initializer()) run(sess,[A,B,mask,C]) but...