Как назначить/обновить подмножество общей переменной тензора в Theano?

При компиляции функции в theano, общая переменная (скажем, X) может быть обновлена, указав updates=[(X, new_value)]. Теперь я пытаюсь обновить только подмножество общей переменной:

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", я полагаю, что цели обновления не могут быть не разделяемыми переменными. Итак, есть ли способ скомпилировать функцию, чтобы просто разрешить подмножество общих переменных?

Ответ 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 FAQ: http://deeplearning.net/software/theano/tutorial/faq_tutorial.html

Ответ 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]

И вот описание об общих переменных в официальном учебнике.

Пожалуйста, спросите, если у вас есть дополнительные вопросы!