TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.experimental.substrates.numpy.math.linalg

View source on GitHub

Functions for common linear algebra operations.

Functions

cholesky_concat(...): Concatenates chol @ chol.T with additional rows and columns.

fill_triangular(...): Creates a (batch of) triangular matrix from a vector of inputs.

fill_triangular_inverse(...): Creates a vector from a (batch of) triangular matrix.

lu_matrix_inverse(...): Computes a matrix inverse given the matrix's LU decomposition.

lu_reconstruct(...): The inverse LU decomposition, X == lu_reconstruct(*tf.linalg.lu(X)).

lu_reconstruct_assertions(...): Returns list of assertions related to lu_reconstruct assumptions.

lu_solve(...): Solves systems of linear eqns A X = RHS, given LU factorizations.

matrix_rank(...): DEPRECATED FUNCTION

pinv(...): DEPRECATED FUNCTION

pivoted_cholesky(...): Computes the (partial) pivoted cholesky decomposition of matrix.

sparse_or_dense_matmul(...): Returns (batched) matmul of a SparseTensor (or Tensor) with a Tensor.

sparse_or_dense_matvecmul(...): Returns (batched) matmul of a (sparse) matrix with a column vector.