code: def scale_logits(self, logits, embeddings):
      embedding_norms = torch.norm(embeddings, p=2, dim=1)
      return logits * embedding_norms.unsqueeze(1) * self.scale
  
imports: import torch
  from .large_margin_softmax_loss import LargeMarginSoftmaxLoss
label_desc: Use image node_id 1 for calling the SphereFaceLoss obj's underlying member method code with example usage: obj.scale_logits(logits, embeddings) and returns: unknown