code: def cross_entropy_1d(
      vocab_logits: torch.Tensor,
      labels: torch.Tensor,
      ignore_index: int = -100,
      process_group: ProcessGroup = None,
  ) -> torch.Tensor:
      return DistCrossEntropy.apply(
          vocab_logits, labels, ignore_index, process_group
      )
  
file_path: ColossalAI/colossalai/shardformer/layer/loss.py
imports: import torch
  import torch.distributed
  from torch.autograd import Function
  from torch.distributed import ProcessGroup
label_desc: Use image node_id 1 for calling a global function with example usage: cross_entropy_1d(vocab_logits, labels, ignore_index, process_group) and returns: DistCrossEntropy