Gradient of Softmax Cross-Entropy Loss - Coding problem | OneNoughtOne