TensorPlay
Reference guides
Copy
View MarkdownDownload .md

tensorplay.linalg

Matrix Properties

tensorplay.linalg.norm

tensorplay.linalg.vector_norm

tensorplay.linalg.matrix_norm

tensorplay.diagonal

tensorplay.linalg.det

tensorplay.linalg.slogdet

tensorplay.linalg.cond

tensorplay.linalg.matrix_rank

Decompositions

tensorplay.linalg.cholesky

tensorplay.linalg.qr

tensorplay.linalg.polar

tensorplay.linalg.lu

tensorplay.linalg.lu_factor

tensorplay.linalg.eig

tensorplay.linalg.eigvals

tensorplay.linalg.eigh

tensorplay.linalg.eigvalsh

tensorplay.linalg.svd

tensorplay.linalg.svdvals

Solvers

tensorplay.linalg.solve

tensorplay.linalg.solve_triangular

tensorplay.linalg.lu_solve

tensorplay.linalg.lstsq

Inverses

tensorplay.linalg.inv

tensorplay.linalg.pinv

Matrix Functions

tensorplay.linalg.matrix_exp

Square matrix exponential via Pade approximation with scaling and squaring (the algorithm of torch's linalg_matrix_exp, Higham 2005).

tensorplay.linalg.matrix_sqrth

Matrix square root via the Denman-Beavers fixed-point iteration (converges for matrices with no eigenvalues on the closed negative real axis).

tensorplay.linalg.matrix_power

Matrix Products

tensorplay.linalg.cross

tensorplay.linalg.matmul

tensorplay.linalg.vecdot

tensorplay.linalg.multi_dot

tensorplay.linalg.householder_product

Tensor Operations

tensorplay.linalg.tensorinv

tensorplay.linalg.tensorsolve

Solves A X = B where A and B are (tuples of) matrices interpreted as a single square system over the trailing dimensions.

Misc

tensorplay.linalg.vander

Experimental Functions

tensorplay.linalg.cholesky_ex

tensorplay.linalg.inv_ex

tensorplay.linalg.solve_ex

tensorplay.linalg.lu_factor_ex

tensorplay.linalg.ldl_factor

tensorplay.linalg.ldl_factor_ex

tensorplay.linalg.ldl_solve

tensorplay.library

tensorplay.library.custom_op

tensorplay.nn

tensorplay.nn.modules.module.Module

Ask DeepWiki