How can I access the model Graph structure?

Can I somehow figure out from a Module what the input layer and output layer are?
i.e. if I have some Layer in hand, can I infer what the input tensors are? I would like to have this information to compute the jacobian of the layer outputs w.r.t layer inputs.
autograd builds some kind of graph during the forward pass, so can I use this somehow?

1 Like