|
10 | 10 |
|
11 | 11 | <meta name="viewport" content="width=device-width, initial-scale=1.0"> |
12 | 12 |
|
13 | | - <title>torch._lobpcg — PyTorch 1.12 documentation</title> |
| 13 | + <title>torch._lobpcg — PyTorch master documentation</title> |
14 | 14 |
|
15 | 15 |
|
16 | 16 |
|
|
40 | 40 | <link rel="index" title="Index" href="../../genindex.html" /> |
41 | 41 | <link rel="search" title="Search" href="../../search.html" /> |
42 | 42 |
|
| 43 | +<!-- |
| 44 | + Search engines should not index the master version of documentation. |
| 45 | + Stable documentation are built without release == 'master'. |
| 46 | +--> |
| 47 | +<meta name="robots" content="noindex"> |
| 48 | + |
43 | 49 |
|
44 | 50 | <!-- Google Analytics --> |
45 | 51 |
|
|
207 | 213 | <div class="pytorch-left-menu-search"> |
208 | 214 |
|
209 | 215 | <div class="version"> |
210 | | - <a href='https://pytorch.org/docs/versions.html'>1.12 ▼</a> |
| 216 | + <a href='https://pytorch.org/docs/versions.html'>master (1.12.0a0+git77665e9 ) ▼</a> |
211 | 217 | </div> |
212 | 218 |
|
213 | 219 |
|
|
227 | 233 |
|
228 | 234 |
|
229 | 235 |
|
| 236 | +<div> |
| 237 | + <a style="color:#F05732" href="https://pytorch.org/docs/stable/_modules/torch/_lobpcg.html"> |
| 238 | + You are viewing unstable developer preview docs. |
| 239 | + Click here to view docs for latest stable release. |
| 240 | + </a> |
| 241 | +</div> |
| 242 | + |
230 | 243 |
|
231 | 244 |
|
232 | 245 |
|
@@ -1346,10 +1359,9 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre> |
1346 | 1359 | <span class="n">SBS</span> <span class="o">=</span> <span class="n">_utils</span><span class="o">.</span><span class="n">qform</span><span class="p">(</span><span class="n">B</span><span class="p">,</span> <span class="n">S</span><span class="p">)</span> |
1347 | 1360 | <span class="n">d_row</span> <span class="o">=</span> <span class="n">SBS</span><span class="o">.</span><span class="n">diagonal</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="o">-</span><span class="mi">2</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span><span class="p">)</span> <span class="o">**</span> <span class="o">-</span><span class="mf">0.5</span> |
1348 | 1361 | <span class="n">d_col</span> <span class="o">=</span> <span class="n">d_row</span><span class="o">.</span><span class="n">reshape</span><span class="p">(</span><span class="n">d_row</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="mi">1</span><span class="p">)</span> |
| 1362 | + <span class="c1"># TODO use torch.linalg.cholesky_solve once it is implemented</span> |
1349 | 1363 | <span class="n">R</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">linalg</span><span class="o">.</span><span class="n">cholesky</span><span class="p">((</span><span class="n">SBS</span> <span class="o">*</span> <span class="n">d_row</span><span class="p">)</span> <span class="o">*</span> <span class="n">d_col</span><span class="p">,</span> <span class="n">upper</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span> |
1350 | | - <span class="n">Id</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">eye</span><span class="p">(</span><span class="n">R</span><span class="o">.</span><span class="n">size</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">),</span> <span class="n">dtype</span><span class="o">=</span><span class="n">R</span><span class="o">.</span><span class="n">dtype</span><span class="p">,</span> <span class="n">device</span><span class="o">=</span><span class="n">R</span><span class="o">.</span><span class="n">device</span><span class="p">)</span> |
1351 | | - <span class="n">Rinv</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">triangular_solve</span><span class="p">(</span><span class="n">Id</span><span class="p">,</span> <span class="n">R</span><span class="p">,</span> <span class="n">upper</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span><span class="o">.</span><span class="n">solution</span> |
1352 | | - <span class="k">return</span> <span class="n">Rinv</span> <span class="o">*</span> <span class="n">d_col</span> |
| 1364 | + <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">linalg</span><span class="o">.</span><span class="n">solve_triangular</span><span class="p">(</span><span class="n">R</span><span class="p">,</span> <span class="n">d_row</span><span class="o">.</span><span class="n">diag_embed</span><span class="p">(),</span> <span class="n">upper</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">left</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span> |
1353 | 1365 |
|
1354 | 1366 | <span class="k">def</span> <span class="nf">_get_svqb</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> |
1355 | 1367 | <span class="n">U</span><span class="p">:</span> <span class="n">Tensor</span><span class="p">,</span> <span class="c1"># Tensor</span> |
|
0 commit comments