Python separable_conv示例

示例#1

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

 def testSeparableConv(self):
   x = np.random.rand(5, 7, 1, 11)
   y = common_layers.separable_conv(
       tf.constant(x, dtype=tf.float32), 13, (3, 1))
   self.evaluate(tf.global_variables_initializer())
   res = self.evaluate(y)
   self.assertEqual(res.shape, (5, 5, 1, 13))

示例#2

 def testSeparableConv(self):
     x = np.random.rand(5, 7, 1, 11)
     y = common_layers.separable_conv(tf.constant(x, dtype=tf.float32), 13,
                                      (3, 1))
     self.evaluate(tf.global_variables_initializer())
     res = self.evaluate(y)
     self.assertEqual(res.shape, (5, 5, 1, 13))