Hum, this looks weird. Can you open an issue in the PyTorch repo?
I had a quick look at the underlying C code, and it does make the required tensors contiguous, so I’m unsure about the reason for this issue.