tf.nn.max_pool

(value, ksize, strides, padding, name=None)

参数是四个,和卷积很类似:

Args Annotation
第一个参数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]这种形式. padding='VALID'时,无自动填充。padding='SAME'时,自动填充,池化后保持shape不变。
def max_pool_2x2(x):
  return tf.nn.max_pool(x, ksize=[1, 2, 2, 1],
            strides=[1, 2, 2, 1], padding='SAME')


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 概述 max pooling是CNN当中的最大值池化操作,其实用法和卷积很类似 说明 tf.nn.max_pool...
    我是谁的小超人阅读 10,096评论 1 1
  • tf.nn.max_pool(value, ksize, strides, padding, data_forma...
    HabileBadger阅读 5,995评论 0 0
  • CNN on TensorFlow 本文大部分内容均参考于: An Intuitive Explanation o...
    _Randolph_阅读 12,299评论 2 31
  • day4.读采铜《精进》有感 昨天断了日更,原因在于我是个闲不住而且太过随心所欲的人,而且太容易被情绪控制,这...
    5040MrYan阅读 3,276评论 0 0
  • 九月份写的没有发表!!!日期总是不诚实。————————————————————————————————————第...
    InkInk阅读 1,508评论 0 1