code: def assign_average_value(self) -> None:
          
          for name, parameter in self._parameters:
              self._backups[name].copy_(parameter.data)
              parameter.data.copy_(self._shadows[name])
docs: Replace all the parameter values with the averages.
          Save the current parameter values to restore later.