0
Je souhaite partitionner mon jeu de données (10 000 images 50x50 RVB) en deux jeux de données. Quelque chose comme:Torche: tenseur de partition
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)
Peu importe ce que je recherche google, je viens d'obtenir la documentation GitHub de la flamme. Comment puis-je faire ceci?