Keras : How to find indices from a tensor for a particular value similar to numpy.where()

Question

I am searching for a Keras command which is similar to python \"numpy.where()\" command. Basically, my idea is to extract the indices from a tensor. In python I

Answer 1

You should try something like:

from keras import backend as K
value = 5
wh = K.tf.where(K.tf.equal(x,value))

when your backend is tensorflow.

Hope that helps