weighted softmax loss CUDA