计算训练网络的梯度

Question

假设我有一个训练有素的模型，我试图计算它的Jacobin（试图在训练后理解一些数学性质）。我正在尝试使用autograd，如下所示：

from autograd import jacobian
jacobian_pred=jacobian(model.predict)
jacobian_pred(x)

其中x来自我的训练集。会引发错误：

TypeError: object of type 'numpy.float32' has no len()

我该怎么办？

谢谢！