From 4c508c1188318de3935bb13bb0f4c5e99f9df222 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 7 Aug 2019 16:09:31 +0200
Subject: [PATCH] Print shape of tensor in tfmad_checkgradient

---
 tests/test_tfmad.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/tests/test_tfmad.py b/tests/test_tfmad.py
index b045288..c1a28f5 100644
--- a/tests/test_tfmad.py
+++ b/tests/test_tfmad.py
@@ -59,6 +59,7 @@ def test_tfmad_two_stencils():
 @pytest.mark.skipif("NO_TENSORFLOW_TEST" in os.environ, reason="Requires Tensorflow")
 def test_tfmad_gradient_check():
     a, b, out = ps.fields("a, b, out: double[21,13]")
+    print(a.shape)
 
     cont = ps.fd.Diff(a, 0) - ps.fd.Diff(a, 1) - ps.fd.Diff(b, 0) + ps.fd.Diff(b, 1)
     discretize = ps.fd.Discretization2ndOrder(dx=1)
@@ -248,12 +249,12 @@ def test_tfmad_two_outputs():
 
 
 def main():
-    test_tfmad_stencil()
-    test_tfmad_two_stencils()
-    test_tfmad_gradient_check_torch()
+    # test_tfmad_stencil()
+    # test_tfmad_two_stencils()
+    # test_tfmad_gradient_check_torch()
     test_tfmad_gradient_check()
-    test_tfmad_vector_input_data()
-    test_tfmad_two_outputs()
+    # test_tfmad_vector_input_data()
+    # test_tfmad_two_outputs()
 
 
 if __name__ == '__main__':
-- 
GitLab