File: library.rst

package info (click to toggle)
pytorch 1.13.1%2Bdfsg-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 139,252 kB
  • sloc: cpp: 1,100,274; python: 706,454; ansic: 83,052; asm: 7,618; java: 3,273; sh: 2,841; javascript: 612; makefile: 323; xml: 269; ruby: 185; yacc: 144; objc: 68; lex: 44
file content (37 lines) | stat: -rw-r--r-- 851 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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: