@Dwight_Foster
Alright, I’ll try that, I too think its not worth it but this is required in this particular case. Thanks for your time!
@KFrank



 KFrank:

those outputs won’t be usefully differentiable so your model won’t train.


That’s exactly what happened when I tried it, I just wanna try a few things and see if I can work around it.



 KFrank:

As an aside, you don’t need the softmax() before the “maximum
index” (argmax()) to get the label prediction. softmax() doesn’t
change the order of its arguments, so you can leave it out.


I am doing that to get the index of the maximum value. Maybe I didn’t understand what you said?