diff options
Diffstat (limited to 'ot/gromov.py')
| -rw-r--r-- | ot/gromov.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gromov.py b/ot/gromov.py index dc95c74..6544260 100644 --- a/ot/gromov.py +++ b/ot/gromov.py @@ -947,7 +947,7 @@ def pointwise_gromov_wasserstein(C1, C2, p, q, loss_fun, index[0] = generator.choice(len_p, size=1, p=nx.to_numpy(p))
T_index0 = nx.reshape(nx.todense(T[index[0], :]), (-1,))
index[1] = generator.choice(
- len_q, size=1, p=nx.to_numpy(T_index0 / T_index0.sum())
+ len_q, size=1, p=nx.to_numpy(T_index0 / nx.sum(T_index0))
)
if alpha == 1:
|