Hi,
As discussed in the other thread, some of the indices are wrong.
You can run it on the cpu to get a more detailed error message or print the values of the indices and check that they are all correct.