コード例 #1
0
ファイル: sandbox.py プロジェクト: BryceWayne/SEM 
def gen_lepolys_torch(N, x):
    lepolys = {}
    for i in range(N + 3):
        if type(lepoly(i, x)) != np.ndarray:
            lepolys[i] = lepoly(i, x)
        else:
            lepolys[i] = torch.from_numpy(lepoly(i, x))
    return lepolys
コード例 #2
0
def gen_lepolys(N, x):
	lepolys = {}
	for i in range(N+1):
		lepolys[i] = sem.lepoly(i, x)
	return lepolys
コード例 #3
0
ファイル: reconstruct.py プロジェクト: yhong2/sem-test 
def gen_lepolys(N, x):
    lepolys = {}
    for i in range(N):
        lepolys[i] = lepoly(i, x)
    return lepolys