How can I assign/update subset of tensor shared variable in Theano? How can I assign/update subset of tensor shared variable in Theano? numpy numpy

How can I assign/update subset of tensor shared variable in Theano?


Use set_subtensor or inc_subtensor:

from theano import tensor as Tfrom theano import function, sharedimport numpyX = 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]

There's now a page about this in the Theano FAQ: http://deeplearning.net/software/theano/tutorial/faq_tutorial.html


This code should solve your problem:

from theano import tensor as Tfrom theano import function, sharedimport numpyX = 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]

And here is the introduction about shared variables in the official tutorial.

Please ask, if you have further questions!