|
5 | 5 | <head> |
6 | 6 | <meta charset="utf-8" /> |
7 | 7 | <meta name="viewport" content="width=device-width, initial-scale=1.0" /> |
8 | | - <title>cmdstanpy.stanfit.mle — CmdStanPy 1.0.6 documentation</title> |
| 8 | + <title>cmdstanpy.stanfit.mle — CmdStanPy 1.0.7 documentation</title> |
9 | 9 |
|
10 | 10 | <!-- Loaded before other Sphinx assets --> |
11 | 11 | <link href="../../../_static/styles/theme.css?digest=1999514e3f237ded88cf" rel="stylesheet"> |
|
56 | 56 | <div id="navbar-start"> |
57 | 57 |
|
58 | 58 | <!-- This will display the version of the docs --> |
59 | | -<a class='navbar-brand' href='../../../index.html'>CmdStanPy 1.0.6</a> |
| 59 | +<a class='navbar-brand' href='../../../index.html'>CmdStanPy 1.0.7</a> |
60 | 60 |
|
61 | 61 | </div> |
62 | 62 |
|
@@ -387,12 +387,17 @@ <h1>Source code for cmdstanpy.stanfit.mle</h1><div class="highlight"><pre> |
387 | 387 | <span class="n">dims</span> <span class="o">=</span> <span class="p">(</span><span class="n">num_rows</span><span class="p">,)</span> <span class="o">+</span> <span class="bp">self</span><span class="o">.</span><span class="n">_metadata</span><span class="o">.</span><span class="n">stan_vars_dims</span><span class="p">[</span><span class="n">var</span><span class="p">]</span> |
388 | 388 | <span class="c1"># pylint: disable=redundant-keyword-arg</span> |
389 | 389 | <span class="k">if</span> <span class="n">num_rows</span> <span class="o">></span> <span class="mi">1</span><span class="p">:</span> |
390 | | - <span class="n">result</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_all_iters</span><span class="p">[:,</span> <span class="n">col_idxs</span><span class="p">]</span><span class="o">.</span><span class="n">reshape</span><span class="p">(</span><span class="n">dims</span><span class="p">,</span> <span class="n">order</span><span class="o">=</span><span class="s1">'F'</span><span class="p">)</span> |
| 390 | + <span class="n">result</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_all_iters</span><span class="p">[:,</span> <span class="n">col_idxs</span><span class="p">]</span> |
391 | 391 | <span class="k">else</span><span class="p">:</span> |
392 | | - <span class="n">result</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_mle</span><span class="p">[</span><span class="n">col_idxs</span><span class="p">]</span><span class="o">.</span><span class="n">reshape</span><span class="p">(</span><span class="n">dims</span><span class="p">[</span><span class="mi">1</span><span class="p">:],</span> <span class="n">order</span><span class="o">=</span><span class="s2">"F"</span><span class="p">)</span> |
| 392 | + <span class="n">result</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_mle</span><span class="p">[</span><span class="n">col_idxs</span><span class="p">]</span> |
| 393 | + <span class="n">dims</span> <span class="o">=</span> <span class="n">dims</span><span class="p">[</span><span class="mi">1</span><span class="p">:]</span> |
393 | 394 |
|
394 | 395 | <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_metadata</span><span class="o">.</span><span class="n">stan_vars_types</span><span class="p">[</span><span class="n">var</span><span class="p">]</span> <span class="o">==</span> <span class="n">BaseType</span><span class="o">.</span><span class="n">COMPLEX</span><span class="p">:</span> |
395 | | - <span class="n">result</span> <span class="o">=</span> <span class="n">result</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="mi">0</span><span class="p">]</span> <span class="o">+</span> <span class="mi">1</span><span class="n">j</span> <span class="o">*</span> <span class="n">result</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="mi">1</span><span class="p">]</span> |
| 396 | + <span class="n">result</span> <span class="o">=</span> <span class="n">result</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="p">::</span><span class="mi">2</span><span class="p">]</span> <span class="o">+</span> <span class="mi">1</span><span class="n">j</span> <span class="o">*</span> <span class="n">result</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="mi">1</span><span class="p">::</span><span class="mi">2</span><span class="p">]</span> |
| 397 | + <span class="n">dims</span> <span class="o">=</span> <span class="n">dims</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span> |
| 398 | + |
| 399 | + <span class="n">result</span> <span class="o">=</span> <span class="n">result</span><span class="o">.</span><span class="n">reshape</span><span class="p">(</span><span class="n">dims</span><span class="p">,</span> <span class="n">order</span><span class="o">=</span><span class="s1">'F'</span><span class="p">)</span> |
| 400 | + |
396 | 401 | <span class="k">return</span> <span class="n">result</span> |
397 | 402 |
|
398 | 403 | <span class="k">else</span><span class="p">:</span> <span class="c1"># scalar var</span> |
|
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&size=40){kind=avatar}
0 commit comments