예제 #1
0
파일: nn_test.py 프로젝트: vismay93/jax 
 def testGluValue(self):
     val = nn.glu(jnp.array([1.0, 0.0]))
     self.assertAllClose(val, jnp.array([0.5]))
예제 #2
0
 def testGluValue(self):
   val = nn.glu(np.array([1.0, 0.0]))
   self.assertAllClose(val, np.array([0.5]), check_dtypes=True)