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