Making Trainable FIR Filter in Deep Neural Networks

Hi,

does anyone have an idea about performing trainable filters in deep learning frameworks like PyTorch? we are working on a motor imagery BCI paradigm and we want to find the best frequency band for the filter to extract ERD/ERS features. Basically, I mean that I want the network to find the best frequency band during the training phase.