Apparently I did not test with large enough images to trigger the bug with torch.narrow on MPS
Attach a file by drag & drop or click to upload