Exemplo n.º 1
0
 def f(x):
   return lax.sin(x) * lax.cos(lax.stop_gradient(x))
Exemplo n.º 2
0
 def f2(x, y):
   return lax.sin(x) * lax.cos(y)
Exemplo n.º 3
0
def cos(x):
    x, = _promote_to_result_dtype(onp.sin, x)
    return lax.cos(x)
Exemplo n.º 4
0
 def _cos(x):
     return lax.cos(x)
Exemplo n.º 5
0
 def f(x):
   return lax.cos(lax.sin(x))
Exemplo n.º 6
0
 def g(x):
   if x > 0:
     return lax.sin(x), 3.
   else:
     return lax.cos(x), 4.