GlobalAvgPool3D

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

Global average 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 NCDHW. For other layouts shape is permuted accordingly.

Outputs:

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