I’m not sure I understand what you are trying to achieve.
Yet, IMHO, the best way to make sure that autograd does what it is supposed to, is to run the code in debug mode, track changes and see what has been updated via BackProbagation. I usually follow this approach to be confident about what is happening. You can run this debugging-test using a simply toy-example that mimics what you are trying to do.