Hi Junpeng,
I think I want the output to have a shape of (2, 1167, 5). Will have to think about this a bit more…
Cheers, Jo