Using softmax loss layer in gradient net (without masking)