def test_high_d_pool(self):
xval = np.random.random((10, 6, 18, 14))
x = Val(xval)
xpooled = AvgPool((2, 2), axis=(2, 1))(x)
self.assertEqual(xpooled.eval().shape, (10, 3, 9, 14))
xpooled = AvgPool((2, 3), axis=(1, 2))(x)
self.assertEqual(xpooled.eval().shape, (10, 3, 6, 14))
xpooled = AvgPool((2, 3), axis=(2, 1))(x)
self.assertEqual(xpooled.eval().shape, (10, 2, 9, 14))
def test_high_d_pool(self):
xval = np.random.random((10, 6, 18, 14))
x = Val(xval)
xpooled = AvgPool((2, 2), axis=(2, 1))(x)
self.assertEqual(xpooled.eval().shape, (10, 3, 9, 14))
xpooled = AvgPool((2, 3), axis=(1, 2))(x)
self.assertEqual(xpooled.eval().shape, (10, 3, 6, 14))
xpooled = AvgPool((2, 3), axis=(2, 1))(x)
self.assertEqual(xpooled.eval().shape, (10, 2, 9, 14))
def test_normal_pool(self):
xval = np.random.random((10, 6))
x = Val(xval)
xpooled = AvgPool((2, 2))(x)
self.assertEqual(xpooled.eval().shape, (5, 3))
def test_normal_pool(self):
xval = np.random.random((10, 6))
x = Val(xval)
xpooled = AvgPool((2, 2))(x)
self.assertEqual(xpooled.eval().shape, (5, 3))