pylops_mpi.proximal.MPIProxOperator

class pylops_mpi.proximal.MPIProxOperator(prox)

MPI-enabled PyProximal Proximal Operator

Common interface for applying (separable) proximal operators in a distributed fashion.

In practice, this class provides methods to compute the norm, proximal operator and gradient between any pyproximal.ProxOperator (which must be the same across ranks) and a pylops_mpi.DistributedArray. It internally handles the extraction of the local array from the distributed array and the creation of the output pylops_mpi.DistributedArray.

Parameters:

proxpyproximal.ProxOperator

PyProximal Proximal Operator to wrap.

Methods

__init__(prox)
prox(x, tau, **kwargs)	Proximal operator applied to a vector
proxdual(x, tau, **kwargs)	Dual Proximal operator applied to a vector