mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
44f41f8e39
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70599 This PR adds `linalg.diagonal` following the Array API: https://data-apis.org/array-api/latest/extensions/linear_algebra_functions.html#linalg-diagonal-x-axis1-0-axis2-1-offset-0 Fixes https://github.com/pytorch/pytorch/issues/62813 cc jianyuh nikitaved pearu mruberry walterddr IvanYashchuk xwang233 Lezcano rgommers pmeier asmeurer leofang AnirudhDagar asi1024 emcastillo kmaehashi Test Plan: Imported from OSS Reviewed By: albanD Differential Revision: D33760506 Pulled By: mruberry fbshipit-source-id: e32c3490321d8c3f31b3bb538bc1f72b39bd2854
107 lines
1.3 KiB
ReStructuredText
107 lines
1.3 KiB
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
torch.linalg
|
|
============
|
|
|
|
Common linear algebra operations.
|
|
|
|
.. 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_factor
|
|
eig
|
|
eigvals
|
|
eigh
|
|
eigvalsh
|
|
svd
|
|
svdvals
|
|
|
|
Solvers
|
|
-------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
solve
|
|
solve_triangular
|
|
lstsq
|
|
|
|
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
|
|
multi_dot
|
|
householder_product
|
|
|
|
Tensor Operations
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
tensorinv
|
|
tensorsolve
|
|
|
|
Experimental Functions
|
|
----------------------
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
cholesky_ex
|
|
inv_ex
|
|
lu_factor_ex
|