Does slice or index of chainer.Variable to get ite

2019-08-19 02:50发布

问题:

Does the following code chainer.Variable still have ability to hold graph and can backward (gradient flow) after slice(a[0,1] or index(a[0]):

>>> a = chainer.Variable(np.array([[1,2,3],[10,11,12]]))
>>> a
variable([[ 1,  2,  3],
          [10, 11, 12]])
>>> a[0]
variable([1, 2, 3])
>>> a[0, 1]
variable([1])

回答1:

Yes. Indexing of chainer.Variable supports backprop.



标签: chainer