One thing that you could try is using the torchviz library to visualize what the backward pass looks like!
torchviz allows you to draw the computation graph of your network. This might be able to help you!