PyTorch/Captum: RuntimeError: gather(): Expected dtype int32/int64 for index

Part of

Runtime Debugging Challenges In Developer Toolchains