Getting around tf.argmax which is not differentiable
tf.argmax is not differentiable because it returns an integer index.  tf.reduce_max and tf.maximum are differentiable
I've written a custom loss function for my neural network but it can't compute any gradients. I thinks it is because I need the index of the highest value and are therefore using argmax to get this index.
As argmax is not differentiable I to get around this but I don't know how it is possible.
Can anyone help?