torch.optim.Optimizer.register\_step\_pre\_hook =============================================== .. currentmodule:: torch.optim .. automethod:: Optimizer.register_step_pre_hook