2017-04-15 108 views
0

我想將我的數據集(10,000個50x50 RGB圖像)分爲兩個數據集。例如:火炬:分區張量

X = torch.rand(10000, 3, 50, 50) 
inds = torch.randperm(X:size(1))[{ { 1, nTrain } }]:long() 
X_selected = X:index(1, inds) 
X_remaining = X:delete(1, inds) 

不管我在谷歌搜索什麼,我只是得到了Torch的GitHub文檔。我怎樣才能做到這一點?

回答

1

你可以試試這個方法

X = torch.rand(10000, 3, 50, 50) 
inds = torch.randperm(X:size(1)):long() 
train_inds = inds:narrow(1, 1, nTrain) 
valid_inds = inds:narrow(1, nTrain + 1, X:size(1) - nTrain) 
X_train = X:index(1, train_inds) 
X_valid = X:index(1, valid_inds)