示例#1
0
def RandomNumpyVector(N):
    v = NumpyVector(N)
    if COMM_WORLD.rank == 0:
        v[:] = _rand(N)
    COMM_WORLD.Bcast(v.content, root=0)
    return v
示例#2
0
def RandomNumpyMatrix(N, M):
    m = NumpyMatrix(N, M)
    if COMM_WORLD.rank == 0:
        m[:, :] = _rand(N, M)
    COMM_WORLD.Bcast(m.content, root=0)
    return m