Pytorch accessing data points from a data batch

Question

I\'m trying to run my model on selected data points in a data batch, but the model generates completely different outputs from directly running on the batch.

Here\'s