Получение меток и классов из данных загрузчиков, но размер файла отличается. ⇐ Python

[Anonymous]

с torch.no_grad(): для i (входные данные, метки) в enumerate(loaders_data['train']): входы = inputs.to(устройство) labels = labels.to(устройство) это мой код, размер пакета — 32, но их длина разная я понимаю это решение

print(len(i)) печать(len(метки)) печать(лен(входные данные)) 32 16 16

но мне нужно набрать 32 за все. можешь мне помочь

Я делаю прогнозы для домашнего задания по машинному обучению. я получаю активации или функции от своей модели.