Torch7：使用切片张量ByteTensor掩模

Question

我有两个张量：

1. 标签是一维张量（5000）
2. 数据集是4D张量（5000,1,32,32）

我想高效地分割与值为1的标签相对应的标签和数据集。我成功切分标签，但不切分数据集。

切片标签：

positive_mask = labels:eq(1) 
sliced_labels = labels[positive_mask] 

我试着做下面的切片数据集和失败：

sliced_dataset = dataset[positive_mask] 
sliced_dataset = dataset[{positive_mask, {}, {}, {}}] 
sliced_dataset = dataset:narrow(1,positive_mask) 
sliced_dataset = dataset:select(1,positive_mask) 

有没有在Torch7执行此一优雅的方法呢？

Answer 1

sliced_dataset = dataset:index(1, positive_mask:nonzero():squeeze())