2D pool (max/avg) output shape: input (batch, channels, height, width), result (batch, channels, out_h, out_w). Channels are preserved.
(batch, channels, height, width)
(batch, channels, out_h, out_w)