What is the recommended way to retrieve losses per sample?

[daMichaelB]

I have a Image-Classification Model and get my predictions for each sample with

preds = trainer.predict(model, datamodule, return_predictions=True)

I am wondering if there is a recommended way to also get the loss-per-sample. As i can not find any docs, i wanted to ask here. Is there something like the following:

preds, losses = trainer.predict(model, datamodule, return_predictions=True, losses=True)

What is the recommended/fastest way to get the losses with lightning-flash ?

Thank you!!