equation class, fix minor bugs, diff domain (#89)

* equation class
* difference domain
* dummy dataloader
* writer class
* refactoring and minor fix
This commit is contained in:
Nicola Demo
2023-05-15 16:06:01 +02:00
parent be11110bb2
commit 0e3625de80
25 changed files with 691 additions and 246 deletions

View File

@@ -8,7 +8,11 @@ def func_vec(x):
return x**2
def func_scalar(x):
return x[:, 0]**2 + x[:, 1]**2 + x[:, 2]**3
print('X')
x_ = x.extract(['x'])
y_ = x.extract(['y'])
mu_ = x.extract(['mu'])
return x_**2 + y_**2 + mu_**3
data = torch.rand((20, 3), requires_grad=True)
inp = LabelTensor(data, ['x', 'y', 'mu'])