Как писать индивидуальные тренировочные петли в керасах с GradientTape