I have a Tensor of size: torch.Size([1, 63840]) which I then unrolled:
torch.Size([1, 63840])
inp_unfolded = inp_seq.unfold(1, 160, 80)
that gives me a