pylops_mpi.DistributedArray.local_split

pylops_mpi.DistributedArray.local_split(global_shape, base_comm, partition, axis)

To get the local shape from the global shape

Parameters:

global_shapetuple

Shape of the global array.

base_commMPI.Comm

Base MPI Communicator.

partitionPartition

Type of partition.

axisint

Axis of distribution

Returns:

local_shapetuple

Shape of the local array.

Examples using pylops_mpi.DistributedArray.local_split

Distributed Array

Distributed Array

Multi-Dimensional Convolution

Multi-Dimensional Convolution

Multi-Dimensional Deconvolution

Multi-Dimensional Deconvolution