I was not able to reproduce the error. I am using torch 1.9.0.post2. Do you still have an issue?
1.9.0.post2