tf.contrib.layers.avg_pool3d
Adds a 3D average pooling op.
tf.contrib.layers.avg_pool3d( inputs, kernel_size, stride=2, padding='VALID', data_format=DATA_FORMAT_NDHWC, outputs_collections=None, scope=None )
It is assumed that the pooling is done per image but not in batch or channels.
Args | |
---|---|
inputs | A 5-D tensor of shape [batch_size, depth, height, width, channels] if data_format is NDHWC , and [batch_size, channels, depth, height, width] if data_format is NCDHW . |
kernel_size | A list of length 3: [kernel_depth, 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 3: [stride_depth, 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. NDHWC (default) and NCDHW 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 NDHWC nor NCDHW . |
© 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/avg_pool3d