oneflow.comm.reduce_scatter

oneflow.comm.reduce_scatter(output, input_list)

Reduces, then scatters a list of tensors to all processes in a group.

Parameters
  • output (Tensor) – Output tensor.

  • input_list (list[Tensor]) – List of tensors to reduce and scatter.