Я работал над проблемой, связанной с базой данных mnist. Я пытался найти значение градиента для каждого пикселя. Для этого я вычислил потерю categoryor_crossentropy между mnist_model.input и target_number (target_number сначала был горячим, а затем преобразован в переменную Keras) и градиентами. Размерность моей функции потерь составляет 28 на 28 на 10, а градиенты составляют 28 на 28 на 1. Я не уверен, как работают коды.
Вот следующий код,
target = to_categorical(target_number)
print(target)
target_variable = K.variable(target)
loss = keras.metrics.categorical_crossentropy(mnist_model.input, target_variable)
gradients = K.gradients(loss, mnist_model.input)
get_grad_values = K.function([mnist_model.input], gradients)