|
193 | 193 | <div class="pytorch-left-menu-search">
|
194 | 194 |
|
195 | 195 | <div class="version">
|
196 |
| - <a href='https://pytorch.org/docs/versions.html'>master (1.11.0a0+git6bc06ec ) ▼</a> |
| 196 | + <a href='https://pytorch.org/docs/versions.html'>master (1.11.0a0+gitce86881 ) ▼</a> |
197 | 197 | </div>
|
198 | 198 |
|
199 | 199 |
|
@@ -480,7 +480,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
480 | 480 | <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="vm">__class__</span><span class="p">()</span>
|
481 | 481 |
|
482 | 482 |
|
483 |
| -<div class="viewcode-block" id="no_grad"><a class="viewcode-back" href="../../../generated/torch.no_grad.html#torch.no_grad">[docs]</a><span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 483 | +<span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
484 | 484 | <span class="sa">r</span><span class="sd">"""Context-manager that disabled gradient calculation.</span>
|
485 | 485 |
|
486 | 486 | <span class="sd"> Disabling gradient calculation is useful for inference, when you are sure</span>
|
@@ -524,7 +524,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
524 | 524 | <span class="n">torch</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
525 | 525 |
|
526 | 526 | <span class="k">def</span> <span class="fm">__exit__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">exc_type</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">exc_value</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">traceback</span><span class="p">:</span> <span class="n">Any</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
527 |
| - <span class="n">torch</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span></div> |
| 527 | + <span class="n">torch</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span> |
528 | 528 |
|
529 | 529 |
|
530 | 530 | <span class="k">class</span> <span class="nc">enable_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span>
|
@@ -570,7 +570,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
570 | 570 | <span class="n">torch</span><span class="o">.</span><span class="n">_C</span><span class="o">.</span><span class="n">_set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span>
|
571 | 571 |
|
572 | 572 |
|
573 |
| -<span class="k">class</span> <span class="nc">set_grad_enabled</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 573 | +<div class="viewcode-block" id="set_grad_enabled"><a class="viewcode-back" href="../../../generated/torch.set_grad_enabled.html#torch.set_grad_enabled">[docs]</a><span class="k">class</span> <span class="nc">set_grad_enabled</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
574 | 574 | <span class="sa">r</span><span class="sd">"""Context-manager that sets gradient calculation to on or off.</span>
|
575 | 575 |
|
576 | 576 | <span class="sd"> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</span>
|
@@ -620,7 +620,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
620 | 620 | <span class="n">torch</span><span class="o">.</span><span class="n">_C</span><span class="o">.</span><span class="n">_set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span>
|
621 | 621 |
|
622 | 622 | <span class="k">def</span> <span class="nf">clone</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
623 |
| - <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="vm">__class__</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">mode</span><span class="p">)</span> |
| 623 | + <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="vm">__class__</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">mode</span><span class="p">)</span></div> |
624 | 624 |
|
625 | 625 |
|
626 | 626 | <span class="k">class</span> <span class="nc">inference_mode</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span>
|
|
0 commit comments