示例#1
0
文件： linalg.py 项目： ahoenselaar/jax 
def solve(a, b):
    a, b = _promote_arg_dtypes(jnp.asarray(a), jnp.asarray(b))
    return lax_linalg._solve(a, b)
示例#2
0
def solve(a, b):
  a, b = _promote_dtypes_inexact(jnp.asarray(a), jnp.asarray(b))
  return lax_linalg._solve(a, b)