Implemented more correct softmax sampling from CDF.