File: linalg.rst

package info (click to toggle)
pytorch-cuda 2.6.0%2Bdfsg-7
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid, trixie
  • size: 161,620 kB
  • sloc: python: 1,278,832; cpp: 900,322; ansic: 82,710; asm: 7,754; java: 3,363; sh: 2,811; javascript: 2,443; makefile: 597; ruby: 195; xml: 84; objc: 68
file content (128 lines) | stat: -rw-r--r-- 1,630 bytes parent folder | download | duplicates (3)
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
.. role:: hidden
    :class: hidden-section

torch.linalg
============

Common linear algebra operations.

See :ref:`Linear Algebra Stability` for some common numerical edge-cases.

.. automodule:: torch.linalg
.. currentmodule:: torch.linalg

Matrix Properties
-----------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    norm
    vector_norm
    matrix_norm
    diagonal
    det
    slogdet
    cond
    matrix_rank

Decompositions
--------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    cholesky
    qr
    lu
    lu_factor
    eig
    eigvals
    eigh
    eigvalsh
    svd
    svdvals

.. _linalg solvers:

Solvers
-------

.. autosummary::
    :toctree: generated
    :nosignatures:

    solve
    solve_triangular
    lu_solve
    lstsq

.. _linalg inverses:

Inverses
--------

.. autosummary::
    :toctree: generated
    :nosignatures:

    inv
    pinv

Matrix Functions
----------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    matrix_exp
    matrix_power

Matrix Products
---------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    cross
    matmul
    vecdot
    multi_dot
    householder_product

Tensor Operations
-----------------

.. autosummary::
    :toctree: generated
    :nosignatures:

    tensorinv
    tensorsolve

Misc
----

.. autosummary::
    :toctree: generated
    :nosignatures:

    vander

Experimental Functions
----------------------
.. autosummary::
    :toctree: generated
    :nosignatures:

    cholesky_ex
    inv_ex
    solve_ex
    lu_factor_ex
    ldl_factor
    ldl_factor_ex
    ldl_solve