Get input params from layer with a jit model

C++
1

Hello,

I would like to have the input parameters of a layer.
For example for a convolution layer conv2D, I would like to have the padding, the stride, etc.

The model I use is a torch::jit::script::Module. I searched the internet and the docs to see where to get all this, but I found nothing.

Thank you very much

2

Maybe you want dump:
https://pytorch.org/cppdocs/api/structtorch_1_1jit_1_1_module.html#_CPPv4NK5torch3jit6Module4dumpEbbb