diff options
author | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-02-02 22:11:04 +0100 |
---|---|---|
committer | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-02-02 22:11:04 +0100 |
commit | 5c00d2dfcf4b0e2835441533f12f195d83652e99 (patch) | |
tree | 1b938487aee9c58cd5fbc2aadbe35896b5e47005 /src/python/doc/rips_complex_tflow_itf_ref.rst | |
parent | c07e645abc27350351af73fa9b24b3d5f881033e (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.rst | 2 |
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]) |