contrib.layers.max_pool2d
tf.contrib.layers.max_pool2d
tf.contrib.layers.max_pool2d
max_pool2d( inputs, kernel_size, stride=2, padding='VALID', data_format=DATA_FORMAT_NHWC, outputs_collections=None, scope=None )
Defined in tensorflow/contrib/layers/python/layers/layers.py
.
See the guide: Layers (contrib) > Higher level ops for building neural network layers
Adds a 2D Max Pooling op.
It is assumed that the pooling is done per image but not in batch or channels.
Args:
-
inputs
: A 4-D tensor of shape[batch_size, height, width, channels]
ifdata_format
isNHWC
, and[batch_size, channels, height, width]
ifdata_format
isNCHW
. -
kernel_size
: A list of length 2: [kernel_height, kernel_width] of the pooling kernel over which the op is computed. Can be an int if both values are the same. -
stride
: A list of length 2: [stride_height, stride_width]. Can be an int if both strides are the same. Note that presently both strides must have the same value. -
padding
: The padding method, either 'VALID' or 'SAME'. -
data_format
: A string.NHWC
(default) andNCHW
are supported. -
outputs_collections
: The collections to which the outputs are added. -
scope
: Optional scope for name_scope.
Returns:
A Tensor
representing the results of the pooling operation.
Raises:
-
ValueError
: Ifdata_format
is neitherNHWC
norNCHW
. -
ValueError
: If 'kernel_size' is not a 2-D list
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/layers/max_pool2d