Обучить нейронную сеть по многомерному временному ряду предсказывать другой временной ряд

Изначально задача математическая. У нас есть какой-то объект, его движение в пространстве можно описать уравнением x' = Ax(t)+Bu(t), A – матрица, характеризующая наш объект, x – положение объекта в момент времени t, B – вектор, показывающий как мы воздействуем на объект, u – управление (воздействие), которое мы оказываем на объект. Я выбираю u, подставляю в систему, t пробегает значения от 0 до 3 секунд, получаю траекторию движения объекта с выбранным управлением. Система трехмерная, то есть у x 3 компоненты. Задача в том, чтобы обучить нейронку из трехмерного временного ряда состояния системы предсказывать временной ряд управления, которое было применено. Файлы с данными для обучения вышлю отдельно (весят больше 100 мб).