class_docstr: Implementation of https://www.ecva.net/papers/eccv_2020/papers_ECCV/papers/123560715.pdf
code: def get_cosine(self, embeddings):
      cosine = self.distance(embeddings, self.W.t())
      cosine = cosine.view(-1, self.num_classes, self.sub_centers)
      cosine, _ = cosine.max(axis=2)
      return cosine
  
file_path: pytorch-metric-learning/src/pytorch_metric_learning/losses/subcenter_arcface_loss.py
imports: import math
  import numpy
  import torch
  from ..utils import common_functions
  from .arcface_loss import ArcFaceLoss
label_desc: Use image node_id 2 for calling the SubCenterArcFaceLoss obj's underlying member method code with example usage: obj.get_cosine(embeddings) and returns: cosine