我正在尝试使用 Theano 来计算函数相对于向量以及几个标量的粗麻布(编辑:也就是说,我本质上希望将标量附加到我正在计算粗麻布的向量上) . 这是一个最小的例子:
import theano
import theano.tensor as T
A = T.vector('A')
b,c = T.scalars('b','c')
y = T.sum(A)*b*c
我的第一次尝试是:
hy = T.hessian(y,[A,b,c])
哪个失败了AssertionError: tensor.hessian expects a (list of) 1 dimensional variable as 'wrt'
我的第二次尝试是将 A、b 和 c 与:
wrt = T.concatenate([A,T.stack(b,c)])
hy = T.hessian(y,[wrt])
哪个失败了DisconnectedInputError: grad method was asked to compute the gradient with respect to a variable that is not part of the computational graph of the cost, or is used only by a non-differentiable operator: Join.0
在这种情况下计算粗麻布的正确方法是什么?
更新:为了澄清我在寻找什么,假设 A 是一个 2 元素向量。那么 Hessian 将是:
[[d2y/d2A1, d2y/dA1dA2, d2y/dA1dB, d2y/dA1dC],
[d2y/dA2dA1, d2y/d2A2, d2y/dA2dB, d2y/dA2dC],
[d2y/dBdA1, d2y/dBdA2, d2y/d2B, d2y/dABdC],
[d2y/dCdA1, d2y/dCdA2, d2y/dCdB, d2y/d2C]]
对于示例函数,它y
应该是:
[[0, 0, C, B],
[0, 0, C, B],
[C, C, 0, A1+A2],
[B, B, A1+A2, 0]]
因此,如果我们要定义一个函数:
f = theano.function([A,b,c], hy)
然后,假设我们可以hy
成功计算,我们会期望输出:
f([1,1], 4, 5) =
[[0, 0, 5, 4],
[0, 0, 5, 4],
[5, 5, 0, 2],
[4, 4, 2, 0]]
在我的实际应用中,A有25个元素,y
比较复杂,但是思路是一样的。