예제 #1
0
파일: linalg.py 프로젝트: ahoenselaar/jax 
def eig(a):
    a = _promote_arg_dtypes(jnp.asarray(a))
    return lax_linalg.eig(a, compute_left_eigenvectors=False)
예제 #2
0
파일: linalg.py 프로젝트: ahoenselaar/jax 
def eigvals(a):
    return lax_linalg.eig(a,
                          compute_left_eigenvectors=False,
                          compute_right_eigenvectors=False)[0]
예제 #3
0
def eig(a):
  a, = _promote_dtypes_inexact(jnp.asarray(a))
  return lax_linalg.eig(a, compute_left_eigenvectors=False)