How would I intercept the first call and do you mind explaining what would happen in the case I want to do Autograd?
I am also seeing a case where the forward function calls torch.flatten(x) but I don’t see the flatten
Thanks for your help!