* support softmax for large columns * more tests
* try * fix bugs * code clean * support mask in softmax * code clean * check for shapes
