code: def forward(self, logit_part, label_part):
      return DistCrossEntropyFunc.apply(logit_part, label_part)
  
imports: import math
  from typing import Callable
  import torch
  from torch import distributed
  from torch.nn.functional import linear, normalize
label_desc: Use image node_id 2 for calling the DistCrossEntropy obj's underlying member method code with example usage: obj.forward(logit_part, label_part) and returns: DistCrossEntropyFunc