Yes, the input of the CrossEntropyLoss is the output of your model.
CrossEntropyLoss works exactly as it would if it expected a (N, C) target. It’s just more efficient to store and carry around the index of the class, and convert the index to one-hot inside the loss function.