Skip to content

Commit f6a2b0e

Browse files
authored
Merge branch 'master' into fix_typo
2 parents 332b114 + 4be0618 commit f6a2b0e

File tree

3 files changed

+22
-8
lines changed

3 files changed

+22
-8
lines changed

advanced_source/torch_script_custom_ops.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1029,5 +1029,5 @@ visible to TorchScript:
10291029
10301030
>>> import torch
10311031
>>> torch.ops.load_library("warp_perspective.so")
1032-
>>> print(torch.ops.custom.warp_perspective)
1032+
>>> print(torch.ops.my_ops.warp_perspective)
10331033
<built-in method custom::warp_perspective of PyCapsule object at 0x7ff51c5b7bd0>

index.rst

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ Welcome to PyTorch Tutorials
5959
.. customcarditem::
6060
:header: Introduction to PyTorch on YouTube
6161
:card_description: An introduction to building a complete ML workflow with PyTorch. Follows the PyTorch Beginner Series on YouTube.
62-
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.PNG
62+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
6363
:link: beginner/introyt.html
6464
:tags: Getting-Started
6565

@@ -362,31 +362,38 @@ Welcome to PyTorch Tutorials
362362
.. customcarditem::
363363
:header: Registering a Dispatched Operator in C++
364364
:card_description: The dispatcher is an internal component of PyTorch which is responsible for figuring out what code should actually get run when you call a function like torch::add.
365-
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.PNG
365+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
366366
:link: advanced/dispatcher.html
367367
:tags: Extending-PyTorch,Frontend-APIs,C++
368368

369369
.. customcarditem::
370370
:header: Extending Dispatcher For a New Backend in C++
371371
:card_description: Learn how to extend the dispatcher to add a new device living outside of the pytorch/pytorch repo and maintain it to keep in sync with native PyTorch devices.
372-
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.PNG
372+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
373373
:link: advanced/extend_dispatcher.html
374374
:tags: Extending-PyTorch,Frontend-APIs,C++
375375

376376
.. customcarditem::
377377
:header: Custom Function Tutorial: Double Backward
378378
:card_description: Learn how to write a custom autograd Function that supports double backward.
379-
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.PNG
379+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
380380
:link: intermediate/custom_function_double_backward_tutorial.html
381381
:tags: Extending-PyTorch,Frontend-APIs
382382

383383
.. customcarditem::
384384
:header: Custom Function Tutorial: Fusing Convolution and Batch Norm
385385
:card_description: Learn how to create a custom autograd Function that fuses batch norm into a convolution to improve memory usage.
386-
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.PNG
386+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
387387
:link: intermediate/custom_function_conv_bn_tutorial.html
388388
:tags: Extending-PyTorch,Frontend-APIs
389389

390+
.. customcarditem::
391+
:header: Forward-mode Automatic Differentiation
392+
:card_description: Learn how to use forward-mode automatic differentiation.
393+
:image: _static/img/thumbnails/cropped/generic-pytorch-logo.png
394+
:link: intermediate/forward_ad_usage.html
395+
:tags: Frontend-APIs
396+
390397
.. Model Optimization
391398
392399
.. customcarditem::
@@ -746,6 +753,7 @@ Additional Resources
746753
:caption: Frontend APIs
747754

748755
intermediate/memory_format_tutorial
756+
intermediate/forward_ad_usage
749757
advanced/cpp_frontend
750758
advanced/torch-script-parallelism
751759
advanced/cpp_autograd

forward_ad_usage.py renamed to intermediate_source/forward_ad_usage.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,17 @@
11
# -*- coding: utf-8 -*-
22
"""
3-
Forward-mode Automatic Differentiation
4-
======================================
3+
Forward-mode Automatic Differentiation (Beta)
4+
=============================================
55
66
This tutorial demonstrates how to use forward-mode AD to compute
77
directional derivatives (or equivalently, Jacobian-vector products).
88
9+
The tutorial below uses some APIs only available in versions >= 1.11
10+
(or nightly builds).
11+
12+
Also note that forward-mode AD is currently in beta. The API is
13+
subject to change and operator coverage is still incomplete.
14+
915
Basic Usage
1016
--------------------------------------------------------------------
1117
Unlike reverse-mode AD, forward-mode AD computes gradients eagerly

0 commit comments

Comments
 (0)