def testGroupNorm(self): x = np.random.rand(5, 7, 3, 16) with self.test_session() as session: y = common_layers.group_norm(tf.constant(x, dtype=tf.float32)) session.run(tf.global_variables_initializer()) res = session.run(y) self.assertEqual(res.shape, (5, 7, 3, 16))
def testGroupNorm(self): x = np.random.rand(5, 7, 3, 16) y = common_layers.group_norm(tf.constant(x, dtype=tf.float32)) self.evaluate(tf.global_variables_initializer()) res = self.evaluate(y) self.assertEqual(res.shape, (5, 7, 3, 16))
def testGroupNorm(self): x = np.random.rand(5, 7, 3, 16) y = common_layers.group_norm(tf.constant(x, dtype=tf.float32)) self.evaluate(tf.global_variables_initializer()) res = self.evaluate(y) self.assertEqual(res.shape, (5, 7, 3, 16))