Consistently implementOneSoftmax #214
Description
OneSoftmax should be implemented as shown in https://www.evanmiller.org/attention-is-off-by-one.html. This is also related to the changes in aeacd0e.
{{ message }}
Consistently implement