diff options
author | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-04-12 15:21:02 +0200 |
---|---|---|
committer | MathieuCarriere <mathieu.carriere3@gmail.com> | 2022-04-12 15:21:02 +0200 |
commit | 27f8df308e3ed935e4ef9f62d23717efebdf36ae (patch) | |
tree | 9fbe0bc93b8b4479af03198bb3400cc697750432 /src/python/gudhi | |
parent | 92517a85ef7d28f4738a27ea850eed9d8c407334 (diff) |
fix doc + reshape in cubical
Diffstat (limited to 'src/python/gudhi')
-rw-r--r-- | src/python/gudhi/tensorflow/cubical_layer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/tensorflow/cubical_layer.py b/src/python/gudhi/tensorflow/cubical_layer.py index 369b0e54..31c44205 100644 --- a/src/python/gudhi/tensorflow/cubical_layer.py +++ b/src/python/gudhi/tensorflow/cubical_layer.py @@ -72,5 +72,5 @@ class CubicalLayer(tf.keras.layers.Layer): if min_pers >= 0: finite_dgm = self.dgms[idx_dim] persistent_indices = tf.where(tf.math.abs(finite_dgm[:,1]-finite_dgm[:,0]) > min_pers) - self.dgms[idx_dim] = tf.gather(finite_dgm, indices=persistent_indices) + self.dgms[idx_dim] = tf.reshape(tf.gather(finite_dgm, indices=persistent_indices), [-1,2]) return self.dgms |