Tf.where doesn't evaluate
Tf.where doesn't evaluate
sess = tf.InteractiveSession()
t = tf.expand_dims(tf.constant(list(range(9))), axis=1)
tf.where(t == 5).eval()
InvalidArgumentError (see above for traceback): WhereOp : Unhandled input dimensions: 0
[[Node: Where_16 = Where[T=DT_BOOL, _device="/job:localhost/replica:0/task:0/device:CPU:0"](Where_16/condition)]]
What's going on here? The corresponding code in Numpy with np.where works.
Does this work for you ?
print( tf.where(tf.equal(t,5)).eval() )– Mohan Radhakrishnan
Sep 2 at 15:18
print( tf.where(tf.equal(t,5)).eval() )
Can you post a longer traceback? What happens if you try running with eager execution instead of interactive session?
– Alexandre Passos
Sep 7 at 19:54
1 Answer
1
In your example, you are evaluating tf.where(False) since the == operator is not overloaded for tensors. (More info e.g. here: TensorFlow operator overloading)
tf.where(False)
==
Try:
sess = tf.InteractiveSession()
t = tf.expand_dims(tf.constant(list(range(9))), axis=1)
tf.where(tf.equal(t, 5)).eval()
Thanks for contributing an answer to Stack Overflow!
But avoid …
To learn more, see our tips on writing great answers.
Some of your past answers have not been well-received, and you're in danger of being blocked from answering.
Please pay close attention to the following guidance:
But avoid …
To learn more, see our tips on writing great answers.
Required, but never shown
Required, but never shown
By clicking "Post Your Answer", you acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.
shouldn't you be doing this inside a session scope?
– DuttaA
Sep 2 at 10:29