Thanks for raising this issue! Since this seems like it could be a possible bug, or at the very least, a migration issue, can you file an issue (essentially this post) over at Issues · pytorch/pytorch · GitHub 1 so that we can take a deeper look?
cc @cbalioglu @H-Huang @Kiuk_Chung @aivanou