@@ -369,7 +369,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
369
369
< span class ="k "> return</ span > < span class ="n "> generator_context</ span >
370
370
371
371
372
- < span class ="k "> class</ span > < span class ="nc "> no_grad</ span > < span class ="p "> (</ span > < span class ="n "> _DecoratorContextManager</ span > < span class ="p "> ):</ span >
372
+ < div class =" viewcode-block " id =" no_grad " > < a class =" viewcode-back " href =" ../../../autograd.html#torch.autograd.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 >
373
373
< span class ="sa "> r</ span > < span class ="sd "> """Context-manager that disabled gradient calculation.</ span >
374
374
375
375
< span class ="sd "> Disabling gradient calculation is useful for inference, when you are sure</ span >
@@ -406,10 +406,10 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
406
406
< 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 "> False</ span > < span class ="p "> )</ span >
407
407
408
408
< 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 ="o "> *</ span > < span class ="n "> args</ span > < span class ="p "> ):</ span >
409
- < 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 >
409
+ < 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 >
410
410
411
411
412
- < 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 >
412
+ < div class ="viewcode-block " id ="enable_grad "> < a class ="viewcode-back " href ="../../../autograd. html#torch.autograd .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 >
413
413
< span class ="sa "> r</ span > < span class ="sd "> """Context-manager that enables gradient calculation.</ span >
414
414
415
415
< span class ="sd "> Enables gradient calculation, if it has been disabled via :class:`~no_grad`</ span >
@@ -448,7 +448,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
448
448
< 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 >
449
449
450
450
451
- < 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 >
451
+ < div class =" viewcode-block " id =" set_grad_enabled " > < a class =" viewcode-back " href =" ../../../autograd.html#torch.autograd.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 >
452
452
< span class ="sa "> r</ span > < span class ="sd "> """Context-manager that sets gradient calculation to on or off.</ span >
453
453
454
454
< span class ="sd "> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</ span >
@@ -493,7 +493,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
493
493
< span class ="k "> pass</ span >
494
494
495
495
< 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 ="o "> *</ span > < span class ="n "> args</ span > < span class ="p "> ):</ span >
496
- < 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 >
496
+ < 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 >
497
497
</ pre > </ div >
498
498
499
499
</ article >
0 commit comments