"register_parametrization()" will register a number of new
     parameters. If such parametrization is registered after the
     optimizer is created, these new parameters will need to be added
     manually to the optimizer. See
     "torch.Optimizer.add_param_group()".
Parameters:
      * module (nn.Module) -- module on which to register the
        parametrization
  * **tensor_name** (*str*) -- name of the parameter or buffer on
    which to register the parametrization

  * **parametrization** (*nn.Module*) -- the parametrization to
    register

Keyword Arguments:
      unsafe (bool) -- a boolean flag that denotes whether the
      parametrization may change the dtype and shape of the tensor.
      Default: False Warning: the parametrization is not checked for
      consistency upon registration. Enable this flag at your own
      risk.
Raises:
      ValueError -- if the module does not have a parameter or a
      buffer named "tensor_name"