Sure! Thank you!
first I was trying to dig in in the maxpooling function and see how it is written and change it, but it was  compiled and couldnot figure it out.
But this should be fine for now, I want to try it on a simple case first.
Just a quick question, when you use the .view, what is the role of * in .view(*input_windows.size()[:-2], -1)?
I haven’t seen * before in view