如何在输入火炬模块之前批量输入？

Question

输入的哪个尺寸应该用于火炬的配料？

我有1000个训练样本，每个训练样例的维度为10 * 5。现在，我想将这些数据作为一组100个示例提供给Sequencer。

我应该如何构建输入？每批输入的尺寸应该是100 * 10 * 5（第一维用于批次）还是10 * 100 * 5（第二维用于批次）？

希望能够找到解释所遵循惯例的相关文件的链接。

容器和模块的约定是否有所改变？

Answer 1

通常是尺寸为100*10*5的张量。如果是图像，则可能需要考虑通道数，因此它应为batchSize*channels*width*height。这样可以轻松访问数据，只需inputs[{i}]即可检索数据。考虑创建另一个Tensor来存储标签（如果使用标签数据）。你可以在这里找到这个例子。 https://github.com/torch/tutorials/blob/master/2_supervised/4_train.lua#L131

我建议您查看教程，在那里您将看到如何在投放网络之前“准备好”数据https://github.com/torch/tutorials