Skip to content

Commit 837f45d

Browse files
committed
auto-generating sphinx docs
1 parent 5a054ab commit 837f45d

File tree

3 files changed

+19
-9
lines changed

3 files changed

+19
-9
lines changed

docs/_modules/torch/nn/functional.html

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -907,7 +907,14 @@ <h1>Source code for torch.nn.functional</h1><div class="highlight"><pre>
907907
<span class="sd"> &gt;&gt;&gt; output = F.nll_loss(F.log_softmax(input), target)</span>
908908
<span class="sd"> &gt;&gt;&gt; output.backward()</span>
909909
<span class="sd"> &quot;&quot;&quot;</span>
910-
<span class="k">return</span> <span class="n">_functions</span><span class="o">.</span><span class="n">thnn</span><span class="o">.</span><span class="n">NLLLoss</span><span class="p">(</span><span class="n">size_average</span><span class="p">,</span> <span class="n">weight</span><span class="o">=</span><span class="n">weight</span><span class="p">)(</span><span class="nb">input</span><span class="p">,</span> <span class="n">target</span><span class="p">)</span></div>
910+
<span class="n">dim</span> <span class="o">=</span> <span class="nb">input</span><span class="o">.</span><span class="n">dim</span><span class="p">()</span>
911+
<span class="k">if</span> <span class="n">dim</span> <span class="o">==</span> <span class="mi">2</span><span class="p">:</span>
912+
<span class="n">f</span> <span class="o">=</span> <span class="n">_functions</span><span class="o">.</span><span class="n">thnn</span><span class="o">.</span><span class="n">NLLLoss</span><span class="p">(</span><span class="n">size_average</span><span class="p">,</span> <span class="n">weight</span><span class="o">=</span><span class="n">weight</span><span class="p">)</span>
913+
<span class="k">elif</span> <span class="n">dim</span> <span class="o">==</span> <span class="mi">4</span><span class="p">:</span>
914+
<span class="n">f</span> <span class="o">=</span> <span class="n">_functions</span><span class="o">.</span><span class="n">thnn</span><span class="o">.</span><span class="n">NLLLoss2d</span><span class="p">(</span><span class="n">size_average</span><span class="p">,</span> <span class="n">weight</span><span class="o">=</span><span class="n">weight</span><span class="p">)</span>
915+
<span class="k">else</span><span class="p">:</span>
916+
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s1">&#39;Expected 2 or 4 dimensions (got </span><span class="si">{}</span><span class="s1">)&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">dim</span><span class="p">))</span>
917+
<span class="k">return</span> <span class="n">f</span><span class="p">(</span><span class="nb">input</span><span class="p">,</span> <span class="n">target</span><span class="p">)</span></div>
911918

912919

913920
<div class="viewcode-block" id="kl_div"><a class="viewcode-back" href="../../../nn.html#torch.nn.functional.kl_div">[docs]</a><span class="k">def</span> <span class="nf">kl_div</span><span class="p">(</span><span class="nb">input</span><span class="p">,</span> <span class="n">target</span><span class="p">,</span> <span class="n">size_average</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>

docs/_modules/torch/nn/utils/rnn.html

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -539,6 +539,9 @@ <h1>Source code for torch.nn.utils.rnn</h1><div class="highlight"><pre>
539539
<span class="sd"> Returns:</span>
540540
<span class="sd"> a :class:`PackedSequence` object</span>
541541
<span class="sd"> &quot;&quot;&quot;</span>
542+
<span class="k">if</span> <span class="n">lengths</span><span class="p">[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span> <span class="o">&lt;=</span> <span class="mi">0</span><span class="p">:</span>
543+
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;length of all samples has to be greater than 0, &quot;</span>
544+
<span class="s2">&quot;but found an element in &#39;lengths&#39; that is &lt;=0&quot;</span><span class="p">)</span>
542545
<span class="k">if</span> <span class="n">batch_first</span><span class="p">:</span>
543546
<span class="nb">input</span> <span class="o">=</span> <span class="nb">input</span><span class="o">.</span><span class="n">transpose</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
544547

docs/torchvision/models.html

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -556,20 +556,20 @@
556556
<td>20.91</td>
557557
</tr>
558558
<tr class="row-odd"><td>VGG-11</td>
559-
<td>32.14</td>
560-
<td>12.12</td>
559+
<td>30.98</td>
560+
<td>11.37</td>
561561
</tr>
562562
<tr class="row-even"><td>VGG-13</td>
563-
<td>31.04</td>
564-
<td>11.40</td>
563+
<td>30.07</td>
564+
<td>10.75</td>
565565
</tr>
566566
<tr class="row-odd"><td>VGG-16</td>
567-
<td>29.11</td>
568-
<td>10.17</td>
567+
<td>28.41</td>
568+
<td>9.62</td>
569569
</tr>
570570
<tr class="row-even"><td>VGG-19</td>
571-
<td>28.42</td>
572-
<td>9.69</td>
571+
<td>27.62</td>
572+
<td>9.12</td>
573573
</tr>
574574
<tr class="row-odd"><td>SqueezeNet 1.0</td>
575575
<td>41.90</td>

0 commit comments

Comments
 (0)