Давайте не будем ничего обучать, а сразу предскажем. А вдруг уже сразу работает? Напишем некоторую функцию "predict", внутри она будет очень простая -- там будет вызов метода "forward", как раз метод forward у нас написан (он, вот реализован). И если вы передадите туда некоторую переменную X, на выходе у вас будет некоторый prediction (то что мы хотим: одно число). И далее есть некоторый код, который рисует вот этот prediction. Давайте посмотрим что происходит. Тут на самом деле, два графика, синим обозначен groud truth, (то, что мы бы хотели на валидации увидеть), X -- это то, что мы передаём в сеть а Y -- это то что мы бы хотели чтобы сеть вернула, а красными точками обозначено то, что сеть нам предсказала. Нетрудно догадаться, что, так как у нас сеть была инициирована случайными числами (то есть, когда вы задаёте слои, они инициализируются некоторыми случайными числами), то на выходе у нас получилась некоторая случайная кривая (она может быть разная в зависимости от запуска).