diff options
author | MathieuCarriere <mathieu.carriere3@gmail.com> | 2021-11-01 14:36:11 +0100 |
---|---|---|
committer | MathieuCarriere <mathieu.carriere3@gmail.com> | 2021-11-01 14:36:11 +0100 |
commit | c4269eef025d4e6c7a763cd99b5dada647693c1d (patch) | |
tree | 802bb37fac936d817ea074aaadcb277b0d41c24a /src/python/doc/rips_complex_tflow_itf_ref.rst | |
parent | 10be82856aee6eb7f4e704757b70c9dab6fe28b8 (diff) |
fix doc
Diffstat (limited to 'src/python/doc/rips_complex_tflow_itf_ref.rst')
-rw-r--r-- | src/python/doc/rips_complex_tflow_itf_ref.rst | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/python/doc/rips_complex_tflow_itf_ref.rst b/src/python/doc/rips_complex_tflow_itf_ref.rst index 15ba4c8e..7300eba0 100644 --- a/src/python/doc/rips_complex_tflow_itf_ref.rst +++ b/src/python/doc/rips_complex_tflow_itf_ref.rst @@ -10,7 +10,7 @@ TensorFlow layer for Vietoris-Rips persistence Example of gradient computed from Vietoris-Rips persistence ----------------------------------------------------------- -.. code-block:: python +.. testcode:: from gudhi.tensorflow import * import numpy as np @@ -23,9 +23,15 @@ Example of gradient computed from Vietoris-Rips persistence with tf.GradientTape() as tape: dgm = rl.call(X) loss = tf.math.reduce_sum(tf.square(.5*(dgm[:,1]-dgm[:,0]))) + grads = tape.gradient(loss, [X]) print(grads[0].numpy()) +.. testoutput:: + + [[-0.5 -0.5] + [ 0.5 0.5]] + Documentation for RipsLayer --------------------------- |