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.