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