revert softmax change, probability distribution seems to have smooted out on RNTNs, making note here: may need to work out row maxes in softmax function as opposed to columns, may have been cause of previous problems