Visualize autograd in pytorch

1

Hi,
Could someone please provide an example snippet code to show how to visualize the autogrd flow, or what happens after loss.backward()
Hopefully in tensorboard. THanks.