tf.contrib.layers.avg_pool2d()

tf.contrib.layers.avg_pool2d(*args, **kwargs)

Adds a 2D average pooling op.

It is assumed that the pooling is done per image but not in batch or channels.

Args:
  • inputs: A Tensor of size [batch_size, height, width, channels].
  • 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'.
  • 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.

doc_TensorFlow
2016-10-14 13:05:20
Comments
Leave a Comment

Please login to continue.