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