Python group_norm示例

示例#1

 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))

示例#2

文件： common_layers_test.py 项目： qixiuai/tensor2tensor

 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))

示例#3

 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))