概述

max pooling是CNN當中的最大值池化操作，其實用法和卷積很類似

說明

tf.nn.max_pool(value, ksize, strides, padding, name=None)

參數

• value：需要池化的輸入，一般池化層接在卷積層后面，所以輸入通常是feature map，依然是[batch, height, width, channels]這樣的shape
• ksize：池化窗口的大小，取一個四維向量，一般是[1, height, width, 1]，因為我們不想在batch和channels上做池化，所以這兩個維度設為了1
• strides：和卷積類似，窗口在每一個維度上滑動的步長，一般也是[1, stride,stride, 1]
• padding：和卷積類似，可以取'VALID' 或者'SAME'
• use_cudnn_on_gpu：bool類型，是否使用cudnn加速，默認為true
• name：指定該操作的name

返回

返回一個Tensor，類型不變，shape仍然是[batch, height, width, channels]這種形式

實例

示例源碼：
假設有這樣一張圖，雙通道
通道1

通道2

import tensorflow as tf

a = tf.constant([[
            [[1., 17.],
             [2., 18.], 
             [3., 19.],
             [4., 20.]],
            [[5., 21.],
             [6., 22.],
             [7., 23.],
             [8., 24.]],
            [[9., 25.],
             [10., 26.],
             [11., 27.],
             [12., 28.]],
            [[13., 29.],
             [14., 30.],
             [15., 31.],
             [16., 32.]]
        ]])
pooling = tf.nn.max_pool(a, [1, 2, 2, 1], [1, 1, 1, 1], padding='VALID')
with tf.Session() as sess:
    print('image: ')
    print(sess.run(a))
    print('\n')
    print('result: ')
    print(sess.run(pooling))

image:
[[[[ 1. 17.]
[ 2. 18.]
[ 3. 19.]
[ 4. 20.]]

[[ 5. 21.]
[ 6. 22.]
[ 7. 23.]
[ 8. 24.]]

[[ 9. 25.]
[ 10. 26.]
[ 11. 27.]
[ 12. 28.]]

[[ 13. 29.]
[ 14. 30.]
[ 15. 31.]
[ 16. 32.]]]]

result:
[[[[ 6. 22.]
[ 7. 23.]
[ 8. 24.]]

[[ 10. 26.]
[ 11. 27.]
[ 12. 28.]]

[[ 14. 30.]
[ 15. 31.]
[ 16. 32.]]]]

通道1

通道2

可以改變步長

pooling = tf.nn.max_pool(a, [1, 2, 2, 1], [1, 2, 2, 1], padding='VALID')

輸出結果：
result:
[[[[ 6. 22.]
[ 8. 24.]]

[[ 14. 30.]
[ 16. 32.]]]]

通道1

通道2