Created by: xhuanlc

The current softmax layer, i.e. paddle.fluid.layers.softmax, only computes softmax along the last dimension. Can you guys add and an "axis" argument to this layer? so the dimension along which to compute softmax can be assigned by users.