be48528b06
* refactor softmax * fix fa * fix mask shape * format * add comments * Remove whitespace
* refactor softmax * fix fa * fix mask shape * format * add comments * Remove whitespace