pytorch/docs/source/fx.rst
Ansley Ussery 080a097935 Add docstring for Proxy (#50145)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50145

Test Plan: Imported from OSS

Reviewed By: pbelevich

Differential Revision: D25854281

Pulled By: ansley

fbshipit-source-id: d7af6fd6747728ef04e86fbcdeb87cb0508e1fd8
2021-01-11 13:47:55 -08:00

33 lines
434 B
ReStructuredText

.. 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:
.. autoclass:: torch.fx.Proxy