[Cherry-pick 2.0.1] Properly set #samples passed to encoder (#3204)#3239

Merged

mthrok merged 1 commit intopytorch:release/2.0from mthrok:cherry-pick-3204

Apr 5, 2023

+61-15

Commits on Apr 4, 2023

Properly set #samples passed to encoder (pytorch#3204 )
mthrok
committed