Customized loss function in PyTorch which uses DNN outputs and additional variables

Question

(I am sorry if my English is not good)

I can create my own loss function in PyTorch if the function requires only DNN output vector(predicted) and DNN output vector(groun