tensorflow如何计算模型梯度？
tensorflow框架，想求分类模型model对输入x的梯度。
在输出层为sigmoid函数下，使用以下代码没有问题。但如果输出层使用softmax函数，该如何求输出对x的梯度?
def grad(x, model):
x = tf.constant([x], dtype=tf.float32)
with tf.GradientTape() as tape:
tape.watch(x)
y_pred = model(x)
gradient = tape.gradient(y_pred, x)
return gradient[0].numpy()感谢回答。
回答
和开发者交流更多问题细节吧，去 写回答
相关文章
相似问题
相关问答用户
请输入您想邀请的人
腾讯云开发者
