Skip to content

Latest commit

 

History

History
37 lines (26 loc) · 851 Bytes

library.rst

File metadata and controls

37 lines (26 loc) · 851 Bytes
Raw

Torch Library API

The PyTorch C++ API provides capabilities for extending PyTorch's core library of operators with user defined operators and data types. Extensions implemented using the Torch Library API are made available for use in both the PyTorch eager API as well as in TorchScript.

For a tutorial style introduction to the library API, check out the Extending TorchScript with Custom C++ Operators tutorial.

Macros

.. doxygendefine:: TORCH_LIBRARY


.. doxygendefine:: TORCH_LIBRARY_IMPL

Classes

.. doxygenclass:: torch::Library
  :members:


.. doxygenclass:: torch::CppFunction
  :members:

Functions

.. doxygengroup:: torch-dispatch-overloads
  :content-only:


.. doxygengroup:: torch-schema-overloads
  :content-only: