How do I get that loss, though, without the softmax inputs?