code: def forward(self, logits, labels):
      loss = multilabel_categorical_crossentropy(labels, logits)
      loss = loss.mean()
      return loss
  
file_path: DeepKE/src/deepke/relation_extraction/document/losses.py
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 2 for calling the ATLoss obj's underlying member method code with example usage: obj.forward(logits, labels) and returns: loss