Python 取模型预测对输入的导数

我正在用tensorflow做一个模型，我需要找到预测对输入向量的导数。例如，如果输入是一个点的（x，y）坐标，预测是该点的温度，我想求温度对x和y的导数。这不能使用Keras的输入类来完成，因为它假设输入是常量而不是变量。如何将输入定义为变量而不是常量