Search code examples
chainer

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

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])
Solution

  • Yes. Indexing of chainer.Variable supports backprop.