Add additional validation check for softmax function that outputs should sum to 1.0