Esempio n. 1
0
def sinc(x):
    _check_arraylike("sinc", x)
    x, = _promote_dtypes_inexact(x)
    eq_zero = lax.eq(x, _lax_const(x, 0))
    pi_x = lax.mul(_lax_const(x, np.pi), x)
    safe_pi_x = _where(eq_zero, _lax_const(x, 1), pi_x)
    return _where(eq_zero, _sinc_maclaurin(0, pi_x),
                  lax.div(lax.sin(safe_pi_x), safe_pi_x))
Esempio n. 2
0
 def f(x, y):
   z = lax.mul(x, y)
   w = lax.sin(z)
   u = lax_internal._reduce_sum(w, [0])
   return (u,)
Esempio n. 3
0
 def f(x):
   return lax.sin(x) * lax.cos(lax.stop_gradient(x))
Esempio n. 4
0
 def f2(x, y):
   return lax.sin(x) * lax.cos(y)
Esempio n. 5
0
def sin(x):
    x, = _promote_to_result_dtype(onp.sin, x)
    return lax.sin(x)
Esempio n. 6
0
 def f(x):
   return lax.cos(lax.sin(x))
Esempio n. 7
0
 def g(x):
   return lax.sin(lax.sin(x))
Esempio n. 8
0
 def g(x):
   if x > 0:
     return lax.sin(x), 3.
   else:
     return lax.cos(x), 4.