Hi, I have a model
class Model(nn.Module):
    def __init__(self):
        pass
    
    def forward(self, x):
        x1 = x * 2 
        x2 = x1 + x 
        x3 = x2 + 1
        return x3

I  want to get x1, x2 and x3.
As a universal method, I CAN NOT return x1, x2, x3. I try forward_hook but it can only trace the module which defined in __init__.