Hi Alex, thanks for the answer!
It is because the upper level will zip() my inputs with another list. But I think you are right! I have never thought that zip() could directly work on a tensor (indeed it works)! Thanks a lot!