I followed https://discuss.pytorch.org/t/typeerror-dataset-object-does-not-support-indexing/72799/19 for creating the custom datasets, target, and indices. I considered indices for the test dataset and its shuffled version separately, which are called idx_test, idx_testsh. For using torch.unique I put all these indices in one list, but it gives this error:
TypeError: _unique(): argument 'input' (position 1) must be Tensor, not list

How can I apply the output of torch.unique over predictions?