How to apply 4D maxpool in pytorch?

PyTorch Live
6

I think, I should use kernel_size = (5,4) instead of (5,4,14).