How to apply 4D maxpool in pytorch?

[ptrblck]

@r00bi

conv2d, kernel size = (5,4), stride(5,1) → output = (32, 1, 3000, 4): torch.nn.Conv2d(5, 5, kernel_size=(5, 4),stride=(5, 1), bias=True)

This layer won’t work as the kernel_size is specified as (5, 4) while the input seems to have a shape of [batch_size, channels, 3000, 1] which is too small in the width.