Fix RAFT input dimension check#8851

NicolasHug merged 3 commits intopytorch:mainpytorch/vision:mainfrom

NicolasHug:aefjnalefjnaljefNicolasHug/vision:aefjnalefjnaljefCopy head branch name to clipboard