首页 > 其他 > 详细

TF-池化函数 tf.nnmax_pool 介绍

时间:2018-06-04 12:42:43      阅读:178      评论:0      收藏:0      [点我收藏+]
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 介绍

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!