Using Autograd .backward() function to calculate an intermediate value in the forward pass of Pytorch model

Question

Hello I am new to Pytorch. I have a simple pytorch module where the output of the module is scalar loss function that depends on the derivative of some polynomial functions. Let