y_true would be the label tensor in the current batch (which you are currently not storing) and y_pred would be the prediction of your model.
For a multi-class classification you can get the prediction via: pred = torch.argmax(output, 1).
Let me know, if you need more information.