|
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.10.0a0+git6862dee ) ▼</a> |
| 196 | + <a href='https://pytorch.org/docs/versions.html'>master (1.10.0a0+git58df55d ) ▼</a> |
197 | 197 | </div>
|
198 | 198 |
|
199 | 199 |
|
@@ -523,7 +523,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
523 | 523 | <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>
|
524 | 524 |
|
525 | 525 |
|
526 |
| -<span class="k">class</span> <span class="nc">enable_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 526 | +<div class="viewcode-block" id="enable_grad"><a class="viewcode-back" href="../../../generated/torch.enable_grad.html#torch.enable_grad">[docs]</a><span class="k">class</span> <span class="nc">enable_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
527 | 527 | <span class="sa">r</span><span class="sd">"""Context-manager that enables gradient calculation.</span>
|
528 | 528 |
|
529 | 529 | <span class="sd"> Enables gradient calculation, if it has been disabled via :class:`~no_grad`</span>
|
@@ -563,10 +563,10 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
563 | 563 | <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="kc">True</span><span class="p">)</span>
|
564 | 564 |
|
565 | 565 | <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>
|
566 |
| - <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> |
| 566 | + <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></div> |
567 | 567 |
|
568 | 568 |
|
569 |
| -<div class="viewcode-block" id="set_grad_enabled"><a class="viewcode-back" href="../../../generated/torch.autograd.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="nb">object</span><span class="p">):</span> |
| 569 | +<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="nb">object</span><span class="p">):</span> |
570 | 570 | <span class="sa">r</span><span class="sd">"""Context-manager that sets gradient calculation to on or off.</span>
|
571 | 571 |
|
572 | 572 | <span class="sd"> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</span>
|
|
0 commit comments