示例#1
0
 def f(x):
   return lax.sin(x) * lax.cos(lax.stop_gradient(x))
示例#2
0
 def f2(x, y):
   return lax.sin(x) * lax.cos(y)
示例#3
0
def cos(x):
    x, = _promote_to_result_dtype(onp.sin, x)
    return lax.cos(x)
示例#4
0
文件： cos.py 项目： gglin001/onnx-jax 
 def _cos(x):
     return lax.cos(x)
示例#5
0
 def f(x):
   return lax.cos(lax.sin(x))
示例#6
0
 def g(x):
   if x > 0:
     return lax.sin(x), 3.
   else:
     return lax.cos(x), 4.