pytorch：从dataset删除数据

2021-03-15 15:45发布

站内问答 / Python
2048 2 3

index = [1,2,3,....],   train_tensor.shape = torch.size(128,30,100),   train_label.shape=torch.size(... 

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中的数据删除怎么办？