I would also like to know if it is possible to get the output of a certain variable inside the forward pass such as variable z = x + self.layer1(x)? Can it be incorporated into the above loop or do I have to get it via some other method?