silvester:

I searched through and realized this is a bug with higher version of Pytorch.


The linked issue is from early-mid 2018 and seems to deal with PyTorch 0.4.
Could you post a code snippet to reproduce this issue in the latest stable version (1.5)?