Calling backward() with indexing in the function

[rfelixmg (Rafa)]

see this answer in this post