def test_avg_pool_same_padding_class_with_inferred_shapes(self): x = np.ones((2, 3, 6)) result = pool.AvgPool(3, 1, padding="SAME", channel_axis=None)(x) np.testing.assert_equal(result.shape, x.shape) # Since x is constant, its avg value should be itself. np.testing.assert_equal(result, x)
def test_avg_pool_same_padding_class(self): x = np.ones((2, 3, 6)) window_shape = [1, 3, 3] strides = [1, 1, 1] avg_pool = pool.AvgPool( window_shape=window_shape, strides=strides, padding="SAME") result = avg_pool(x) np.testing.assert_equal(result.shape, x.shape) # Since x is constant, its avg value should be itself. np.testing.assert_equal(result, x)