토치 : 파티션 텐서

Question

내 데이터 세트 (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) 

Google 검색이 무엇이든, 나는 Torch의 GitHub 문서를 얻습니다. 어떻게해야합니까?

Answer 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) 

을 시도 할 수 있습니다