Hello, now I have a pytorch model, how can I get the relationship between each layer of the model as shown in the arrow above, this picture is just an example, the relationship between layers is not accurate, how can I get the correct relationship?