sigpy.linop.AllReduce

class sigpy.linop.AllReduce(shape, comm, in_place=False)[source]

Performs all reduce between MPI ranks.

Returns input if in_place is True. Otherwise returns reduced array on the input device.

Parameters:
  • shape (tuple of ints) – Input/output shape.
  • comm (Communicator) – Communicator.
__init__(shape, comm, in_place=False)[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(shape, comm[, in_place]) Initialize self.
apply(input) Apply linear operation on input.

Attributes

H Return adjoint linear operator.
N Return normal linear operator.