torch.nn.utils.parametrize.register\_parametrization ==================================================== .. currentmodule:: torch.nn.utils.parametrize .. autofunction:: register_parametrization