Validation of allocated output tensor shapes#339

NicolasHug merged 5 commits intometa-pytorch:mainmeta-pytorch/torchcodec:mainfrom

NicolasHug:shape_assertNicolasHug/torchcodec:shape_assertCopy head branch name to clipboard