GlobalMaxPool3D

class GlobalMaxPool3D(layout='NCDHW', **kwargs)

Global max pooling operation for 3D data (spatial or spatio-temporal).

Parameters:

layout : str, default ‘NCDHW’

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

Inputs:

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

Outputs:

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