Pytorch model structure

How to obtain the input-output relation between the model layer and the layer

Could you speak a bit more to what you’re looking for, maybe with an example?

Now that I have a pytorch model, how do I code the input and output of each layer, just like bottom and top in caffe

Now I’m doing TensorRT to speed up the Pytorch model, and you need to get the input and output relationships of each layer of the model and the parameters of each layer, thus manually building this