mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
d453b3c4d4
This was long-due, as it keeps comming up in issues. Fixes https://github.com/pytorch/pytorch/issues/85950 Fixes https://github.com/pytorch/pytorch/issues/59720 Fixes https://github.com/pytorch/pytorch/issues/59782 Pull Request resolved: https://github.com/pytorch/pytorch/pull/88313 Approved by: https://github.com/soumith, https://github.com/mruberry
129 lines
1.6 KiB
ReStructuredText
129 lines
1.6 KiB
ReStructuredText
.. 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
|