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

Update test_operators.py

Browse Source
This commit is contained in:
Nicola Demo
2022-07-20 17:59:01 +02:00
parent d6adce507b
commit 62f203fcc3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_operators.py
View File

@@ -33,9 +33,9 @@ def test_grad_vector_output():
def test_div_vector_output():
grad_tensor_v = div(tensor_v, inp)
assert grad_tensor_v.shape == (20, 3)
assert grad_tensor_v.shape == (20, 1)
grad_tensor_v = div(tensor_v, inp, components=['a', 'b'], d=['x', 'mu'])
assert grad_tensor_v.shape == (inp.shape[0], 2)
assert grad_tensor_v.shape == (inp.shape[0], 1)
def test_nabla_scalar_output():
laplace_tensor_v = nabla(tensor_s, inp, components=['a'], d=['x', 'y'])