예제 #1
0
파일: djax_test.py 프로젝트: yashk2810/jax 
 def f(x):
   return sin(x)
예제 #2
0
파일: djax_test.py 프로젝트: yashk2810/jax 
 def f(x):
   y = sin(x)
   z = sin(y)
   return add(y, z)
예제 #3
0
파일: djax_test.py 프로젝트: yashk2810/jax 
 def f(x):
   y = sin(x)
   return reduce_sum(y, axes=(0,))