def testMaxPool(self): y = extensions.max_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="MAX", padding="VALID", strides=[2, 3], ))
def testMaxPool(self): y = extensions.max_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], "MAX", "VALID", strides=[2, 3],))