Я заинтересован в создании моделей обучения подкреплению с простотой API Keras. К сожалению, я не могу извлечь градиент вывода (не ошибка) по отношению к весам. Я нашел следующий код, который выполняет аналогичную функцию (Карты значимости нейронных сетей (с использованием Keras))

get_output = theano.function([model.layers[0].input],model.layers[-1].output,allow_input_downcast=True)
fx = theano.function([model.layers[0].input] ,T.jacobian(model.layers[-1].output.flatten(),model.layers[0].input), allow_input_downcast=True)
grad = fx([trainingData])

Буду признателен за любые идеи о том, как рассчитать градиент выходных данных модели по отношению к весам для каждого слоя.