Okay guys I was able to fix this by changing the dimensions of the input data. Initially all input vectors had dimensions [0:1] converting them to [-1:1] or [-2:2] fixed the problem. Apparently the summation after softmax does not work correctly for strictly positive data.