шаг 1 (прямой проход) выполнялся инструкцией predictions = model(input), а шаг 2 (получение градиентов, вычисленных с помощью GradientTape) — инструкцией gradients = tape.gradient(loss, model.weights).
Глубокое обучение на Python. 2-е межд. издание
·
Франсуа Шолле
