summaryrefslogtreecommitdiff
path: root/src/python/doc/rips_complex_tflow_itf_ref.rst
diff options
context:
space:
mode:
authorMathieuCarriere <mathieu.carriere3@gmail.com>2021-11-05 00:35:51 +0100
committerMathieuCarriere <mathieu.carriere3@gmail.com>2021-11-05 00:35:51 +0100
commit734622d5a8816cfdaaed2aaa4b9b3212fb6a259c (patch)
tree1b3d1b10b0f28bdb654f1f53fe51d9d96c485af3 /src/python/doc/rips_complex_tflow_itf_ref.rst
parent1597a5b4fc1aec9f825e430e80b2a843a9037043 (diff)
update doc and pieces of code
Diffstat (limited to 'src/python/doc/rips_complex_tflow_itf_ref.rst')
-rw-r--r--src/python/doc/rips_complex_tflow_itf_ref.rst4
1 files changed, 1 insertions, 3 deletions
diff --git a/src/python/doc/rips_complex_tflow_itf_ref.rst b/src/python/doc/rips_complex_tflow_itf_ref.rst
index 7300eba0..bd9c5da5 100644
--- a/src/python/doc/rips_complex_tflow_itf_ref.rst
+++ b/src/python/doc/rips_complex_tflow_itf_ref.rst
@@ -13,11 +13,9 @@ Example of gradient computed from Vietoris-Rips persistence
.. testcode::
from gudhi.tensorflow import *
- import numpy as np
import tensorflow as tf
- Xinit = np.array([[1.,1.],[2.,2.]], dtype=np.float32)
- X = tf.Variable(initial_value=Xinit, trainable=True)
+ X = tf.Variable([[1.,1.],[2.,2.]], dtype=tf.float32, trainable=True)
rl = RipsLayer(maximum_edge_length=2., dimension=0)
with tf.GradientTape() as tape: