Как передать градиенты на вход временного шага 1 в сети LSTM в Tensorflow?
В Tensorflow я загружаю матрицу X входных векторов в сеть LSTM. Первый вектор входной матрицы (временной шаг 1) является вектором с горячим кодированием, тогда как следующие векторы X (следующие временные шаги) являются нулевыми векторами. Теперь я хочу распространить градиенты потерь на входной вектор на временном шаге 1.
Если я использую следующую операцию, я получаю градиенты ввода, но для каждого временного шага.
gradients_input = sess.run(tf.gradients(loss, X), feed_dict={X:batch_x, Y:batch_y})
Любая идея, как я могу вычислить градиенты только для временного шага 1?