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

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