def testBELU(self): x = np.random.rand(5, 2, 1, 12) y = common_layers.belu(tf.constant(x, dtype=tf.float32)) actual = self.evaluate(y) self.assertEqual(actual.shape, (5, 2, 1, 12))
def testBELU(self): x = np.random.rand(5, 2, 1, 12) y = common_layers.belu(tf.constant(x, dtype=tf.float32)) actual = self.evaluate(y) self.assertEqual(actual.shape, (5, 2, 1, 12))
def testBELU(self): with self.test_session() as session: x = np.random.rand(5, 2, 1, 12) y = common_layers.belu(tf.constant(x, dtype=tf.float32)) actual = session.run(y) self.assertEqual(actual.shape, (5, 2, 1, 12))