From b702b2701add90a26080d27bcd528101820b44cf Mon Sep 17 00:00:00 2001 From: Michal Klein <46717574+michalk8@users.noreply.github.com> Date: Wed, 20 Dec 2023 00:43:59 +0100 Subject: [PATCH] Remvoe note --- src/ott/neural/layers.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/ott/neural/layers.py b/src/ott/neural/layers.py index baf219a03..78c2ef3b8 100644 --- a/src/ott/neural/layers.py +++ b/src/ott/neural/layers.py @@ -92,9 +92,6 @@ class PosDefPotentials(nn.Module): matrix. Finally, the :math:`c_i` are contained in a vector of size ``(num_potentials,)``. - .. note:: - The quadratic potential has diagonal + low-rank structure. - Args: num_potentials: Dimension of the output. rank: Rank of the matrices :math:`A_i` used as low-rank factors