Zero grad on single parameter

The gradient will be computed when you call:

a.backward()
1 Like