Hi,
Is it this issue: https://github.com/pytorch/pytorch/issues/36455 5 ?
I think this particular comment on the issue answers your question https://github.com/pytorch/pytorch/issues/36455#issuecomment-614097391 12?