lib.mpi.dot_array

lib.mpi.dot_array(a, b, mpicomm=None)

Return dot product of input arrays a and b. Currently accepts one-dimensional b or two-dimensional a and b. b must be scattered along first axis, hence a scattered along last axis.