tf入門-池化函數 tf.nn.max_pool 的介紹

轉載自此大神 http://blog.csdn.net/mao_xiao_feng/article/details/53453926

 

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

有些地方能夠從卷積去參考【TensorFlow】tf.nn.conv2d是怎樣實現卷積的？ 

 

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'

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