`forward()` works but `backward()` fails on the same batch

Double post from here.