code: def reduce_scatter(
      output, input_list, op=ReduceOp.SUM, group=group.WORLD
  ):
      """
      Reduces, then scatters a list of tensors to all processes in a group.
  
      Arguments:
          output (Tensor): Output tensor.
          input_list (list[Tensor]): List of tensors to reduce and scatter.
          op (optional): One of the values from
              ``torch.distributed.ReduceOp``
              enum.  Specifies an operation used for element-wise reductions.
          group (ProcessGroup, optional): The process group to work on.
  
      Returns:
          Tensor: Output of the collective.
  
      """
      return _Reduce_Scatter.apply(op, group, output, *input_list)
  
imports: import torch
  import torch.distributed
  from torch.autograd import Function
label_desc: Use image node_id 5 for calling a global function with example usage: reduce_scatter(output, input_list, op, group) and returns: _Reduce_Scatter