Torch7：使用 ByteTensor 掩码对张量进行切片答案

【问题标题】：Torch7: Slice Tensor using ByteTensor maskTorch7：使用 ByteTensor 掩码对张量进行切片
【发布时间】：2016-03-30 15:30:49
【问题描述】：

我有两个张量：

标签是一维张量 (5000)
数据集是 4D Tensor (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 中是否有一种优雅的方法来执行此操作？

【问题讨论】：

标签： torch

【解决方案1】：

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

【讨论】：