mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
37 lines
851 B
ReStructuredText
37 lines
851 B
ReStructuredText
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
|
|
<https://pytorch.org/tutorials/advanced/torch_script_custom_ops.html>`_
|
|
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:
|