tfp.substrates.jax.math.hpsd_quadratic_form_solvevec

Computes rhs^T matrix^-1 rhs, where matrix is HPSD.

Given matrix and rhs, computes rhs^T @ matrix^-1 rhs, where matrix is a Hermitian positive semi-definite matrix.

matrix Floating-point Tensor of shape [..., N, N]. Represents a Hermitian positive semi-definite matrix.
rhs Floating-point Tensor of shape [..., N].
cholesky_matrix (Optional) Floating-point Tensor of shape [..., N, N] that represents a Cholesky factor of matrix.

hpsd_quadratic_form_solvevec Scalar Tensor.