sigpy.linop.AllReduce

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

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)

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.