|
230 | 230 | <div class="pytorch-left-menu-search">
|
231 | 231 |
|
232 | 232 | <div class="version">
|
233 |
| - <a href='https://pytorch.org/docs/versions.html'>main (2.1.0a0+git02c2b75 ) ▼</a> |
| 233 | + <a href='https://pytorch.org/docs/versions.html'>main (2.1.0a0+git2e73c86 ) ▼</a> |
234 | 234 | </div>
|
235 | 235 |
|
236 | 236 |
|
@@ -1073,8 +1073,8 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
|
1073 | 1073 |
|
1074 | 1074 | <span class="k">if</span> <span class="n">m</span> <span class="o"><</span> <span class="mi">3</span> <span class="o">*</span> <span class="n">n</span><span class="p">:</span>
|
1075 | 1075 | <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span>
|
1076 |
| - <span class="s2">"LPBPCG algorithm is not applicable when the number of A rows (=</span><span class="si">{}</span><span class="s2">)"</span> |
1077 |
| - <span class="s2">" is smaller than 3 x the number of requested eigenpairs (=</span><span class="si">{}</span><span class="s2">)"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">m</span><span class="p">,</span> <span class="n">n</span><span class="p">)</span> |
| 1076 | + <span class="sa">f</span><span class="s2">"LPBPCG algorithm is not applicable when the number of A rows (=</span><span class="si">{</span><span class="n">m</span><span class="si">}</span><span class="s2">)"</span> |
| 1077 | + <span class="sa">f</span><span class="s2">" is smaller than 3 x the number of requested eigenpairs (=</span><span class="si">{</span><span class="n">n</span><span class="si">}</span><span class="s2">)"</span> |
1078 | 1078 | <span class="p">)</span>
|
1079 | 1079 |
|
1080 | 1080 | <span class="n">method</span> <span class="o">=</span> <span class="s2">"ortho"</span> <span class="k">if</span> <span class="n">method</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">method</span>
|
@@ -1610,9 +1610,7 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
|
1610 | 1610 | <span class="k">assert</span> <span class="n">B</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span>
|
1611 | 1611 | <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span>
|
1612 | 1612 | <span class="s2">"Overdetermined shape of U:"</span>
|
1613 |
| - <span class="s2">" #B-cols(=</span><span class="si">{}</span><span class="s2">) >= #U-cols(=</span><span class="si">{}</span><span class="s2">) + #V-cols(=</span><span class="si">{}</span><span class="s2">) must hold"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span> |
1614 |
| - <span class="n">B</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> <span class="n">U</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> <span class="n">V</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> |
1615 |
| - <span class="p">)</span> |
| 1613 | + <span class="sa">f</span><span class="s2">" #B-cols(=</span><span class="si">{</span><span class="n">B</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><span class="si">}</span><span class="s2">) >= #U-cols(=</span><span class="si">{</span><span class="n">U</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><span class="si">}</span><span class="s2">) + #V-cols(=</span><span class="si">{</span><span class="n">V</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><span class="si">}</span><span class="s2">) must hold"</span> |
1616 | 1614 | <span class="p">)</span>
|
1617 | 1615 | <span class="bp">self</span><span class="o">.</span><span class="n">ivars</span><span class="p">[</span><span class="s2">"ortho_i"</span><span class="p">]</span> <span class="o">=</span> <span class="n">i</span>
|
1618 | 1616 | <span class="bp">self</span><span class="o">.</span><span class="n">ivars</span><span class="p">[</span><span class="s2">"ortho_j"</span><span class="p">]</span> <span class="o">=</span> <span class="n">j</span>
|
|
0 commit comments