symbolics: evaluate transpose on elements of tensors in case of deriv…

…ative
devitocodes · Feb 27, 2023 · 988aad6 · 988aad6
1 parent a737e6c
commit 988aad6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/devito/types/basic.py b/devito/types/basic.py
@@ -677,6 +677,11 @@ def __init_finalize__(self, *args, **kwargs):
     def doit(self, **hint):
         return self
 
+    @property
+    def T(self):
+        new = super().T
+        return new.applyfunc(lambda x: getattr(x, 'T', x))
+
     def _eval_matrix_mul(self, other):
         """
         Copy paste from sympy to avoid explicit call to sympy.Add

diff --git a/tests/test_tensors.py b/tests/test_tensors.py
@@ -149,6 +149,14 @@ def test_vector_transpose(func1):
     assert np.all([f1[i] == f2[i] for i in range(3)])
 
 
+@pytest.mark.parametrize('func1', [VectorFunction, VectorTimeFunction])
+def test_vector_transpose_deriv(func1):
+    grid = Grid(tuple([5]*3))
+    f1 = func1(name="f1", grid=grid)
+    f2 = f1.dx.T
+    assert all([f2[i] == f1[i].dx.T for i in range(3)])
+
+
 @pytest.mark.parametrize('func1', [TensorFunction, TensorTimeFunction,
                                    VectorFunction, VectorTimeFunction])
 def test_tensor_fd(func1):