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
def gen_lepolys(N, x):
lepolys = {}
for i in range(N+1):
lepolys[i] = sem.lepoly(i, x)
return lepolys
def gen_lepolys(N, x):
lepolys = {}
for i in range(N):
lepolys[i] = lepoly(i, x)
return lepolys