码迷,mamicode.com
首页 > 其他好文 > 详细

TF-池化函数 tf.nnmax_pool 介绍

时间:2018-06-04 14:38:05      阅读:265      评论:0      收藏:0      [点我收藏+]

标签:channels   一个   body   span   参数   矩阵   形式   color   class   

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

参数介绍:

value:需要池化的输入,一般池化层接在卷积层后面,所以输入通常是feature map,依然是[batch, height, width, channels]这样的shape。

batch 训练时一个batch的图片数量
in_height 图片高度
in_width 图片宽度
in_channels 图像通道数

ksize:池化窗口的大小,取一个四维向量,一般是[1, in_height, in_width, 1],因为我们不想在batch和channels上做池化,所以这两个维度设为了1。

strides:和卷积类似,窗口在每一个维度上滑动的步长,一般也是[1, stride,stride, 1]。

padding:和卷积类似,可以取‘VALID‘ 或者‘SAME‘。

返回值介绍:

结果返回一个Tensor,这个输出,就是我们常说的feature map,shape仍然是 [batch, height, width, channels] 这种形式。

 

 

特征图的行列数计算公式:

对于n*n的矩阵, 使用 f*f 的核进行卷积, 填充宽度为p, 若纵向步幅为s1, 横向步幅为s2则特征图的行列数为:

   技术分享图片

 

TF-池化函数 tf.nnmax_pool 介绍

标签:channels   一个   body   span   参数   矩阵   形式   color   class   

原文地址:https://www.cnblogs.com/xh_chiang/p/9132240.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!