blockJacobi.py 文件源码

def __init__(self, X, kern, M):
        super(BlockJacobi, self).__init__("BlockJacobi")
        self.M = M

        start = time.time()
        X_split = np.array_split(X, M)
        kern_blocks = np.zeros((M),dtype=object)

        for t in xrange(M):
            size = np.shape(X_split[t])[0]
            kern_blocks[t] = kern.K(X_split[t], X_split[t]) + kern.noise*np.identity(size)

        self.duration = time.time()-start
        self.blocks = kern_blocks
        self.precon = block_diag(*kern_blocks)