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

前端 未结 2 549
野的像风
野的像风 2020-12-14 16:49

When compiling a function in theano, a shared variable(say X) can be updated by specifying updates=[(X, new_value)]. Now I am trying to update only

2条回答
  •  太阳男子
    2020-12-14 17:24

    This code should solve your problem:

    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]
    

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

    Please ask, if you have further questions!

提交回复
热议问题