tf.contrib.layers.max_pool2d
Adds a 2D Max Pooling op.
tf.contrib.layers.max_pool2d( inputs, kernel_size, stride=2, padding='VALID', data_format=DATA_FORMAT_NHWC, outputs_collections=None, scope=None )
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] if data_format is NHWC , and [batch_size, channels, height, width] if data_format is NCHW . |
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) and NCHW 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 | If data_format is neither NHWC nor NCHW . |
ValueError | If 'kernel_size' is not a 2-D list |
© 2020 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/versions/r1.15/api_docs/python/tf/contrib/layers/max_pool2d