Softmax cost is now fixed to account for multidimensional targets