Skip to content

Commit 6619c3b

Browse files
Circle CICircle CI
authored andcommitted
CircleCI update of dev docs (3086).
1 parent 88735fc commit 6619c3b

File tree

278 files changed

+732574
-731443
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

278 files changed

+732574
-731443
lines changed
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
-317 Bytes
592 Bytes
299 Bytes
301 Bytes
-115 Bytes
360 Bytes
-42 Bytes
-170 Bytes

master/_modules/ot/backend.html

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1908,7 +1908,7 @@ <h1>Source code for ot.backend</h1><div class="highlight"><pre>
19081908
<div class="viewcode-block" id="NumpyBackend.trace">
19091909
<a class="viewcode-back" href="../../gen_modules/ot.backend.html#ot.backend.NumpyBackend.trace">[docs]</a>
19101910
<span class="k">def</span><span class="w"> </span><span class="nf">trace</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">):</span>
1911-
<span class="k">return</span> <span class="n">np</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span><span class="n">a</span><span class="p">)</span></div>
1911+
<span class="k">return</span> <span class="n">np</span><span class="o">.</span><span class="n">einsum</span><span class="p">(</span><span class="s2">&quot;...ii&quot;</span><span class="p">,</span> <span class="n">a</span><span class="p">)</span></div>
19121912

19131913

19141914
<div class="viewcode-block" id="NumpyBackend.inv">
@@ -2570,7 +2570,7 @@ <h1>Source code for ot.backend</h1><div class="highlight"><pre>
25702570
<div class="viewcode-block" id="JaxBackend.trace">
25712571
<a class="viewcode-back" href="../../gen_modules/ot.backend.html#ot.backend.JaxBackend.trace">[docs]</a>
25722572
<span class="k">def</span><span class="w"> </span><span class="nf">trace</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">):</span>
2573-
<span class="k">return</span> <span class="n">jnp</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span><span class="n">a</span><span class="p">)</span></div>
2573+
<span class="k">return</span> <span class="n">jnp</span><span class="o">.</span><span class="n">diagonal</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">axis1</span><span class="o">=-</span><span class="mi">2</span><span class="p">,</span> <span class="n">axis2</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span><span class="o">.</span><span class="n">sum</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">)</span></div>
25742574

25752575

25762576
<div class="viewcode-block" id="JaxBackend.inv">
@@ -3352,7 +3352,7 @@ <h1>Source code for ot.backend</h1><div class="highlight"><pre>
33523352
<div class="viewcode-block" id="TorchBackend.trace">
33533353
<a class="viewcode-back" href="../../gen_modules/ot.backend.html#ot.backend.TorchBackend.trace">[docs]</a>
33543354
<span class="k">def</span><span class="w"> </span><span class="nf">trace</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">):</span>
3355-
<span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span><span class="n">a</span><span class="p">)</span></div>
3355+
<span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">diagonal</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">dim1</span><span class="o">=-</span><span class="mi">2</span><span class="p">,</span> <span class="n">dim2</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span><span class="o">.</span><span class="n">sum</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">)</span></div>
33563356

33573357

33583358
<div class="viewcode-block" id="TorchBackend.inv">
@@ -4015,7 +4015,7 @@ <h1>Source code for ot.backend</h1><div class="highlight"><pre>
40154015
<div class="viewcode-block" id="CupyBackend.trace">
40164016
<a class="viewcode-back" href="../../gen_modules/ot.backend.html#ot.backend.CupyBackend.trace">[docs]</a>
40174017
<span class="k">def</span><span class="w"> </span><span class="nf">trace</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">):</span>
4018-
<span class="k">return</span> <span class="n">cp</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span><span class="n">a</span><span class="p">)</span></div>
4018+
<span class="k">return</span> <span class="n">cp</span><span class="o">.</span><span class="n">trace</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">axis1</span><span class="o">=-</span><span class="mi">2</span><span class="p">,</span> <span class="n">axis2</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span></div>
40194019

40204020

40214021
<div class="viewcode-block" id="CupyBackend.inv">

master/_modules/ot/gaussian.html

Lines changed: 442 additions & 69 deletions
Large diffs are not rendered by default.

master/_modules/ot/utils.html

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1524,6 +1524,40 @@ <h1>Source code for ot.utils</h1><div class="highlight"><pre>
15241524

15251525

15261526

1527+
<div class="viewcode-block" id="exp_bures">
1528+
<a class="viewcode-back" href="../../gen_modules/ot.utils.html#ot.exp_bures">[docs]</a>
1529+
<span class="k">def</span><span class="w"> </span><span class="nf">exp_bures</span><span class="p">(</span><span class="n">Sigma</span><span class="p">,</span> <span class="n">S</span><span class="p">,</span> <span class="n">nx</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
1530+
<span class="w"> </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
1531+
<span class="sd"> Exponential map in Bures-Wasserstein space at Sigma:</span>
1532+
1533+
<span class="sd"> .. math::</span>
1534+
<span class="sd"> \exp_\Sigma(S) = (I_d+S)\Sigma(I_d+S).</span>
1535+
1536+
<span class="sd"> Parameters</span>
1537+
<span class="sd"> ----------</span>
1538+
<span class="sd"> Sigma : array-like (d,d)</span>
1539+
<span class="sd"> SPD matrix</span>
1540+
<span class="sd"> S : array-like (d,d)</span>
1541+
<span class="sd"> Symmetric matrix</span>
1542+
<span class="sd"> nx : module, optional</span>
1543+
<span class="sd"> The numerical backend module to use. If not provided, the backend will</span>
1544+
<span class="sd"> be fetched from the input matrices `Sigma, S`.</span>
1545+
1546+
<span class="sd"> Returns</span>
1547+
<span class="sd"> -------</span>
1548+
<span class="sd"> P : array-like (d,d)</span>
1549+
<span class="sd"> SPD matrix obtained as the exponential map of S at Sigma</span>
1550+
<span class="sd"> &quot;&quot;&quot;</span>
1551+
<span class="k">if</span> <span class="n">nx</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
1552+
<span class="n">nx</span> <span class="o">=</span> <span class="n">get_backend</span><span class="p">(</span><span class="n">Sigma</span><span class="p">,</span> <span class="n">S</span><span class="p">)</span>
1553+
<span class="n">d</span> <span class="o">=</span> <span class="n">S</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
1554+
<span class="n">Id</span> <span class="o">=</span> <span class="n">nx</span><span class="o">.</span><span class="n">eye</span><span class="p">(</span><span class="n">d</span><span class="p">,</span> <span class="n">type_as</span><span class="o">=</span><span class="n">S</span><span class="p">)</span>
1555+
<span class="n">C</span> <span class="o">=</span> <span class="n">Id</span> <span class="o">+</span> <span class="n">S</span>
1556+
1557+
<span class="k">return</span> <span class="n">nx</span><span class="o">.</span><span class="n">einsum</span><span class="p">(</span><span class="s2">&quot;ij,jk,kl -&gt; il&quot;</span><span class="p">,</span> <span class="n">C</span><span class="p">,</span> <span class="n">Sigma</span><span class="p">,</span> <span class="n">C</span><span class="p">)</span></div>
1558+
1559+
1560+
15271561
<div class="viewcode-block" id="check_number_threads">
15281562
<a class="viewcode-back" href="../../gen_modules/ot.utils.html#ot.check_number_threads">[docs]</a>
15291563
<span class="k">def</span><span class="w"> </span><span class="nf">check_number_threads</span><span class="p">(</span><span class="n">numThreads</span><span class="p">):</span>

master/_sources/auto_examples/backends/plot_Sinkhorn_gradients.rst.txt

Lines changed: 1 addition & 1 deletion

master/_sources/auto_examples/backends/plot_dual_ot_pytorch.rst.txt

Lines changed: 40 additions & 40 deletions

master/_sources/auto_examples/backends/plot_optim_gromov_pytorch.rst.txt

Lines changed: 2 additions & 2 deletions

0 commit comments

Comments
 (0)