class_docstr: Class implementing Kmeans in PyTorch
code: def forward(self, x):
      # Compute the Euclidean distance
      dist = torch.cdist(
          x, self.centroids, compute_mode="donot_use_mm_for_euclid_dist"
      )
      label = torch.argmin(dist, dim=1)
      return label
  
file_path: hummingbird/hummingbird/ml/operator_converters/sklearn/cluster.py
imports: import torch
  from .._physical_operator import PhysicalOperator
  from onnxconverter_common.registration import register_converter
label_desc: Use image node_id 2 for calling the KMeans obj's underlying member method code with example usage: obj.forward(x) and returns: label