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