Python conv_stride2_multistep示例

示例#1

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

 def testConvStride2MultiStep(self):
   x1 = np.random.rand(5, 32, 16, 11)
   a = common_layers.conv_stride2_multistep(
       tf.constant(x1, dtype=tf.float32), 4, 16)
   self.evaluate(tf.global_variables_initializer())
   actual = self.evaluate(a[0])
   self.assertEqual(actual.shape, (5, 2, 1, 16))

示例#2

 def testConvStride2MultiStep(self):
     x1 = np.random.rand(5, 32, 16, 11)
     a = common_layers.conv_stride2_multistep(
         tf.constant(x1, dtype=tf.float32), 4, 16)
     self.evaluate(tf.global_variables_initializer())
     actual = self.evaluate(a[0])
     self.assertEqual(actual.shape, (5, 2, 1, 16))