Torch7: Dilim tensör ByteTensor iki tansörlerine sahip

Question

maske kullanılarak:

1. etiket 1B tensör olan (5000)
2. veri kümesi 4D tensör (5000,1,32,32)

olduğu

Değer 1'in etiketine karşılık gelen etiketleri ve veri kümesini verimli bir şekilde dilimlemek istiyorum. Etiketleri değil, veri kümesini dilimlemeyi başarabilirim.

etiket dilimleme:

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

ben veri kümesi dilim aşağıdakileri yaparak çalıştı ve başarısız oldu:

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 bu gerçekleştirmek için bir zarif yaklaşım var mı?

Answer 1

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