pytorch：从dataset删除数据

Question

index = [1,2,3,....],

train_tensor.shape = torch.size(128,30,100),

train_label.shape=torch.size(128,1,1)

train_set = dataset(train_tensor,train_label)

我想按index把trainl_set中的数据删除怎么办？

 

Answer 1

那你按剩下的index选数据不就行了：

1.得到你要用的index

2.train_tensor.index_select(0,index),train_label.index_select(0,index)

3.train_set = dataset(train_tensor,train_label)