tf.nn.max_pool 原生的池化操作,而 tf.layers.max_pooling2d是进行了封装,对许多参数进行了设置,使用起来更方便
tf.layers.max_pooling2d:
用于2D输入的最大池化层(例如图像).
参数:
- inputs:池的张量,秩必须为4.
- pool_size:2个整数的整数或元组/列表:(pool_height,pool_width),用于指定池窗口的大小.可以是单个整数,以指定所有空间维度的相同值.
- strides:2个整数的整数或元组/列表,用于指定池操作的步幅.可以是单个整数,以指定所有空间维度的相同值.
- padding:一个字符串,表示填充方法,“valid”或“same”,不区分大小写.
- data_format:一个字符串,表示输入中维度的顺序.支持channels_last(默认)和channels_first;channels_last对应于具有形状(batch, height, width, channels)的输入,而channels_first对应于具有形状(batch, channels, height, width)的输入.
- name:字符串,图层的名称.
tf.nn.max_pool:
在输入上执行最大池化.
参数:
- value:由data_format指定格式的4-D Tensor.
- ksize:具有4个元素的1-D整数Tensor.输入张量的每个维度的窗口大小.
- strides:具有4个元素的1-D整数Tensor.输入张量的每个维度的滑动窗口的步幅.
- padding:一个字符串,可以是'VALID'或'SAME'.填充算法.
- data_format:一个字符串.支持'NHWC','NCHW'和'NCHW_VECT_C'.
- name:操作的可选名称.