Does slice or index of chainer.Variable to get item in chainer has backward ability?

216 Views Asked by At

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

There are 1 best solutions below

3
On BEST ANSWER

Yes. Indexing of chainer.Variable supports backprop.