Print shape of tensor in tfmad_checkgradient

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__':