max_col_norm refactor for Maxout and Softmax