Jacobian Matrix of the Softmax Transformation - Coding problem | OneNoughtOne