torch.nn.utils.parametrize.remove_parametrizations
torch.nn.utils.parametrize.remove_parametrizations(module, tensor_name, leave_parametrized=True)
Removes the parametrizations on a tensor in a module.


If "leave_parametrized=True", "module[tensor_name]" will be set
     to its current output. In this case, the parametrization shall
     not change the "dtype" of the tensor.


If "leave_parametrized=False", "module[tensor_name]" will be set
     to the unparametrised tensor in
     "module.parametrizations[tensor_name].original". This is only
     possible when the parametrization depends on just one tensor.


Parameters:
      * module (nn.Module) -- module from which remove the
        parametrization
  * **tensor_name** (*str*) -- name of the parametrization to be
    removed

  * **leave_parametrized** (*bool**, **optional*) -- leave the
    attribute "tensor_name" parametrized. Default: "True"

Returns: