Use the correct batch values in the output#2191

svekars merged 2 commits intopytorch:mainpytorch/tutorials:mainfrom

spzala:batchdataspzala/tutorials:batchdataCopy head branch name to clipboard