Hi, I am using Integrated Gradients with a NN model that has both sequential (LSTM) and non sequential layers (MLP). I was following the basic steps as given in the examples and ran into a weird error.

Code snippet:

test_tens.requires_grad_()

attr = ig.attribute(test_tens, target=1, return_convergence_delta=False)

attr = attr.detach().numpy()

error:

in line 2 of the above snippet:

" index 1 is out of bounds for dimension 1 with size 1 "

Probably some problem around the target thing, but not sure what to do