code: def init_embed_(self, data):
      if self.initted:
          return
      print("Performing Kemans init for codebook")
      embed, cluster_size = kmeans(
          data, self.num_tokens, 10, use_cosine_sim=True
      )
      self.weight.data.copy_(embed)
      self.cluster_size.data.copy_(cluster_size)
      self.initted.data.copy_(torch.Tensor([True]))
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  import torch.distributed
label_desc: Use image node_id 2 for calling the EmbeddingEMA obj's underlying member method code with example usage: obj.init_embed_(data) without return types