No such function exist.
I am sure it could be implemented easily at the nn.Module level like zero_grad() is done. @smth do you think this is something we want in the core or more of a user specific usage?