folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change default reduction in SystemEquation (#317)

Browse Source 
* Update system_equation.py
* Update test_systemequation.py
This commit is contained in:
luAndre00
2024-08-01 17:47:51 +02:00
committed by GitHub
parent f9316e359a
commit 16261c9baf
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_equations/test_systemequation.py
View File

@@ -39,7 +39,7 @@ def test_residual():
u = torch.pow(pts, 2)
u.labels = ['u1', 'u2']
eq_1 = SystemEquation([eq1, eq2])
eq_1 = SystemEquation([eq1, eq2], reduction='mean')
res = eq_1.residual(pts, u)
assert res.shape == torch.Size([10])
@@ -47,6 +47,10 @@ def test_residual():
res = eq_1.residual(pts, u)
assert res.shape == torch.Size([10])
eq_1 = SystemEquation([eq1, eq2], reduction='none')
eq_1 = SystemEquation([eq1, eq2], reduction=None)
res = eq_1.residual(pts, u)
assert res.shape == torch.Size([10, 3])
eq_1 = SystemEquation([eq1, eq2])
res = eq_1.residual(pts, u)
assert res.shape == torch.Size([10, 3])