lib.mpi.argmax_array

lib.mpi.argmax_array(data, *args, mpicomm=None, axis=None, **kwargs)

Return index of maximum in input array data along axis.