class_docstr: Class implementing Kmeans in PyTorch
code: def __init__(self, logical_operator, centroids, device):
      super(KMeans, self).__init__(logical_operator, regression=True)
  
      self.centroids = torch.nn.Parameter(
          torch.FloatTensor(centroids), requires_grad=False
      )
  
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 1 to create a new KMeans object from inherited base classes: PhysicalOperator, torch.nn with example: obj = KMeans(logical_operator, centroids, device)