def _reshape_to_original(self, param_name: str, value: _torch.Tensor) -> _torch.Tensor:
        """
        Reshape any value flattened for k-means back to original parameter shape.
        """

        Reshape any value flattened for k-means back to original parameter shape.
        
coremltools/optimize/torch/_utils/k_means.py