Python avg_pool示例

示例#1

文件： extensions_test.py 项目： yliu45/trax

 def testAvgPool(self):
   y = extensions.avg_pool(np.ones([5, 320, 480, 3]), [3, 5], [2, 3], "VALID")
   self.assertAllEqual(
       y,
       tf.nn.pool(
           input=tf.ones([5, 320, 480, 3]),
           window_shape=[3, 5],
           pooling_type="AVG",
           padding="VALID",
           strides=[2, 3],
       ))

示例#2

 def testAvgPool(self):
   y = extensions.avg_pool(np.ones([5, 320, 480, 3]), [3, 5], [2, 3], "VALID")
   self.assertAllEqual(
       y, tf.nn.pool(tf.ones([5, 320, 480, 3]), [3, 5], "AVG", "VALID",
                     strides=[2, 3],))