fx.rst

.. 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: