Sync tensordict#3478

voldyman wants to merge 1 commit intometa-pytorch:mainmeta-pytorch/torchrec:mainfrom

voldyman:export-D85157386voldyman/torchrec:export-D85157386Copy head branch name to clipboard