Which conversion module should I use?

Hi, I am trying to use a CNN to obtain features of shape [batch size, context length, hidden size] from a matrix of shape [batch size, context length, utterance length, hidden size], where context length is variable.

I don’t know which conversion module should be used.

Thanks.