def _update_parameters_impl(self, param_name: str, new_value: _torch.Tensor):
        """
        Update the parameter corresponding to this parameter name with the
        new value after reshaping to original parameter shape.
        """

        Update the parameter corresponding to this parameter name with the
        new value after reshaping to original parameter shape.
        
coremltools/optimize/torch/_utils/k_means.py