Hello @ptrblck (and Aoo)!



 ptrblck:




 Aoo_Lmay:

Is there an argmax analog that could be employed?


Yes, softmax (sometimes also called softargmax) is the smooth version of argmax, which provides valid gradients.


As a minor technical clarification, softmax() is a smooth version of
the one_hot() encoding of the result of argmax().
Best.
K. Frank