Should I take indicies before or after converting the input into FloatTensor  ?
if after should I just use locs = Train_y == 0 and Train_negx = Train_x[~locs] ?
if before should I convert locs into Tensor and which type of tensor ?