Question

在theano中编译函数时，可以通过指定updates=[(X, new_value)]来更新共享变量（比如X）。现在我试图只更新共享变量的子集：

from theano import tensor as T
from theano import function
import numpy

X = T.shared(numpy.array([0,1,2,3,4]))
Y = T.vector()
f = function([Y], updates=[(X[2:4], Y)] # error occur:
                                        # 'update target must 
                                        # be a SharedVariable'

代码会引发错误“更新目标必须是SharedVariable”，我猜这意味着更新目标不能是非共享变量。那么有没有办法将函数编译成只是udpate共享变量的子集？

Answer 1

使用set_subtensor或inc_subtensor：

from theano import tensor as T
from theano import function, shared
import numpy

X = shared(numpy.array([0,1,2,3,4]))
Y = T.vector()
X_update = (X, T.set_subtensor(X[2:4], Y))
f = function([Y], updates=[X_update])
f([100,10])
print X.get_value() # [0 1 100 10 4]

Theano常见问题解答中现在有一个关于此内容的页面：http://deeplearning.net/software/theano/tutorial/faq_tutorial.html

Answer 2

此代码可以解决您的问题：

from theano import tensor as T
from theano import function, shared
import numpy

X = shared(numpy.array([0,1,2,3,4], dtype='int'))
Y = T.lvector()
X_update = (X, X[2:4]+Y)
f = function(inputs=[Y], updates=[X_update])
f([100,10])
print X.get_value()
# output: [102 13]

这是introduction about shared variables in the official tutorial。

如果您还有其他问题，请询问！

如何在Theano中分配/更新张量共享变量的子集？

2 个答案: