Let’s say I have the indices which I optained using
locs = Train_y == 0
in numpy I can just do
Train_negx = Train_x[np.invert(locs)]
but in pytorch it is not working, what is the equivalent to this method ? Knowing I’m working on output Variable.
I guess the problem is in the invert version