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/ls_simplex_tree_tflow_itf_ref.rst | |
parent | c07e645abc27350351af73fa9b24b3d5f881033e (diff) |
fixed bugs from the new API
Diffstat (limited to 'src/python/doc/ls_simplex_tree_tflow_itf_ref.rst')
-rw-r--r-- | src/python/doc/ls_simplex_tree_tflow_itf_ref.rst | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/doc/ls_simplex_tree_tflow_itf_ref.rst b/src/python/doc/ls_simplex_tree_tflow_itf_ref.rst index 3200b8e5..56bb4492 100644 --- a/src/python/doc/ls_simplex_tree_tflow_itf_ref.rst +++ b/src/python/doc/ls_simplex_tree_tflow_itf_ref.rst @@ -32,7 +32,7 @@ Example of gradient computed from lower-star filtration of a simplex tree sl = LowerStarSimplexTreeLayer(simplextree=st, dimensions=[0]) with tf.GradientTape() as tape: - dgm = sl.call(F) + dgm = sl.call(F)[0][0] loss = tf.math.reduce_sum(tf.square(.5*(dgm[:,1]-dgm[:,0]))) grads = tape.gradient(loss, [F]) |