I want to create my custom loss function, unfortunately I need the input from multiple previous layers.
Input_x -> conv1 -> relu1 -> conv2 -> relu2 -> conv3 -> relu3 -> conv4 -> relu4 -> output

I attached my custom loss function between relu3 and conv4, but I need to evaluate the input from relu1. Normally forward method accept 2 arguments which are self, and x (output from previous layers). If my understanding correct, this is not supported???
Thank you