softmax and softmax loss layers work across channels
test softmax and softmax with loss across channels
implement GPU version of Softmax