Передача вывода из модели в качестве входного загрузчика данных pytorch

#pytorch

#pytorch

Вопрос:

Я делаю прогнозирование кадров видео с использованием LSTM. Я обучил сеть, используя входной кадр во время (t) и целевой кадр во время (t 1).

Теперь для набора тестов я хочу, чтобы LSTM использовал выходной кадр из time (t) в качестве входного кадра во время (t 1), возможно ли это сделать с помощью загрузчика данных?

Комментарии:

1. да, но вам не обязательно его использовать. Вы пробовали что-то, что не сработало, или что именно вы хотите знать? Все, что я могу сказать: Да

2. Как бы вы это реализовали? Есть ли какие-либо примеры, на которые вы можете мне указать?