Torch：分区张量

【问题标题】：Torch: partition tensorTorch：分区张量
【发布时间】：2017-09-11 20:18:41
【问题描述】：

我想将我的数据集（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)

无论我在 Google 上搜索什么，我都会得到 Torch 的 GitHub 文档。我该怎么做？

【问题讨论】：

标签： lua torch

【解决方案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)

【讨论】：