redistribute#
- caput.mpiarray.MPIArray.redistribute(axis: int) MPIArray[source]#
Change the rank distribution axis of the array.
Guarantees that the result is c-contiguous.
- Parameters:
- axis
int Axis to distribute over.
- axis
- Returns:
- redistributed_array
MPIArray MPIArrayinstance of the original array, distributed over the new axis. If the new axis is the same as the old axis, the original array is returned.
- redistributed_array
- Raises:
MemoryErrorIf there is not enough memory to perform the redistribution.