summaryrefslogtreecommitdiff
path: root/src/python/doc/rips_complex_tflow_itf_ref.rst
diff options
context:
space:
mode:
authorMathieuCarriere <mathieu.carriere3@gmail.com>2022-02-02 22:11:04 +0100
committerMathieuCarriere <mathieu.carriere3@gmail.com>2022-02-02 22:11:04 +0100
commit5c00d2dfcf4b0e2835441533f12f195d83652e99 (patch)
tree1b938487aee9c58cd5fbc2aadbe35896b5e47005 /src/python/doc/rips_complex_tflow_itf_ref.rst
parentc07e645abc27350351af73fa9b24b3d5f881033e (diff)
fixed bugs from the new API
Diffstat (limited to 'src/python/doc/rips_complex_tflow_itf_ref.rst')
-rw-r--r--src/python/doc/rips_complex_tflow_itf_ref.rst2
1 files changed, 1 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 fc42e5c9..104b0971 100644
--- a/src/python/doc/rips_complex_tflow_itf_ref.rst
+++ b/src/python/doc/rips_complex_tflow_itf_ref.rst
@@ -19,7 +19,7 @@ Example of gradient computed from Vietoris-Rips persistence
rl = RipsLayer(maximum_edge_length=2., dimensions=[0])
with tf.GradientTape() as tape:
- dgm = rl.call(X)
+ dgm = rl.call(X)[0][0]
loss = tf.math.reduce_sum(tf.square(.5*(dgm[:,1]-dgm[:,0])))
grads = tape.gradient(loss, [X])