softmax operating on specific dimension.
unit-test for softmax layer along specified dimension
utility to split tensor dimension
Softmax Layer new behavior (CUDA)