[WIP][FX] Add FX page to docs

ghstack-source-id: 908adcc22394195b6795c726c6199af22e61e41e Pull Request resolved: #48814
pytorch · Dec 5, 2020 · 7571fbe · 7571fbe
1 parent 3535f82
commit 7571fbe
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/docs/source/fx.rst b/docs/source/fx.rst
@@ -0,0 +1,30 @@
+.. currentmodule:: torch.fx
+
+torch.fx
+=============
+
+Overview
+--------
+.. automodule:: torch.fx
+
+
+API Reference
+-------------
+
+.. autofunction:: torch.fx.symbolic_trace
+
+.. autoclass:: torch.fx.GraphModule
+  :members:
+
+  .. automethod:: __init__
+
+.. autoclass:: torch.fx.Graph
+  :members:
+
+  .. automethod:: __init__
+
+.. autoclass:: torch.fx.Node
+  :members:
+
+.. autoclass:: torch.fx.Tracer
+  :members:
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -61,6 +61,7 @@ Features described in this documentation are classified by release status:
    torch.distributions <distributions>
    torch.fft <fft>
    futures
+   fx
    torch.hub <hub>
    torch.jit <jit>
    torch.linalg <linalg>