Ejemplo n.º 1
0
 def testGluValue(self):
     val = nn.glu(jnp.array([1.0, 0.0]))
     self.assertAllClose(val, jnp.array([0.5]))
Ejemplo n.º 2
0
 def testGluValue(self):
   val = nn.glu(np.array([1.0, 0.0]))
   self.assertAllClose(val, np.array([0.5]), check_dtypes=True)