GlobalMaxPool2D

class GlobalMaxPool2D(layout='NCHW', **kwargs)

Global max pooling operation for two dimensional (spatial) data.

Parameters:

layout : str, default ‘NCHW’

Dimension ordering of data and out (‘NCHW’ or ‘NHWC’). ‘N’, ‘C’, ‘H’, ‘W’ stands for batch, channel, height, and width dimensions respectively. padding is applied on ‘H’ and ‘W’ dimension.

Inputs:

• data: 4D input tensor with shape (batch_size, in_channels, height, width) when layout is NCHW. For other layouts shape is permuted accordingly.

Outputs:

• out: 4D output tensor with shape (batch_size, channels, 1, 1) when layout is NCHW.