avg_pool2d

tfsnippet.layers.avg_pool2d(*args, **kwargs)

2D average pooling over spatial dimensions.

Parameters:

• input (Tensor) – The input tensor, at least 4-d.
• pool_size (int or (int, int)) – Pooling size over spatial dimensions.
• strides (int or (int, int)) – Strides over spatial dimensions.
• channels_last (bool) – Whether or not the channel axis is the last axis in input? (i.e., the data format is “NHWC”)
• padding – One of {“valid”, “same”}, case in-sensitive.
• name (str) – Default name of the name scope. If not specified, generate one according to the method name.

Returns:

The output tensor.

Return type:

tf.Tensor