How to train networks using gradients when the last operation that produces result is ArgMax?

Hi All,

This is my first post here, but I hope I will stick around.

I have a network that takes a number input values (lets assume ***n***) and produces ***k*** output values.

The output vector can be considered as a distribution for the given set of specific input values.

However, to use this network I need to apply the ArgMax operator - to choose one specific element from the output layer. That is - each y\_i corresponds to some …

networks neuralnetworks