Thanks for the explanation. Just to check it again.
So in keras, the input data is [batch, timestep/length, filter/channel], while in pytorch it becomes [batch, filter/channel, timestep/length] right?
If that’s the case, I guess I should do sth like transpose((0,2,1)).