{ }
View source on GitHub |
Computes the eigenvalues of a Hermitian tridiagonal matrix.
tf.linalg.eigh_tridiagonal(
alpha,
beta,
eigvals_only=True,
select='a',
select_range=None,
tol=None,
name=None
)
Returns | |
---|---|
eig_vals
|
The eigenvalues of the matrix in non-decreasing order. |
eig_vectors
|
If eigvals_only is False the eigenvectors are returned in
the second output argument.
|
Raises | |
---|---|
ValueError
|
If input values are invalid. |
NotImplemented
|
Computing eigenvectors for eigvals_only = False is
not implemented yet.
|
This op implements a subset of the functionality of scipy.linalg.eigh_tridiagonal.
Add support for outer batch dimensions.
Examples
import numpy
eigvals = tf.linalg.eigh_tridiagonal([0.0, 0.0, 0.0], [1.0, 1.0])
eigvals_expected = [-numpy.sqrt(2.0), 0.0, numpy.sqrt(2.0)]
tf.assert_near(eigvals_expected, eigvals)
# ==> True