Pytorch如何增加批量大小答案

【问题标题】：Pytorch how to increase batch sizePytorch如何增加批量大小
【发布时间】：2020-11-08 21:24:35
【问题描述】：

我目前有一个 torch.Size([1, 3, 256, 224]) 的张量，但我需要它作为输入形状 [32, 3, 256, 224]。我正在实时捕获数据，因此数据加载器似乎不是一个好的选择。有什么简单的方法可以获取 32 个大小的 torch.Size([1, 3, 256, 224]) 并将它们组合起来以创建 1 个大小为 [32, 3, 256, 224] 的张量？

【问题讨论】：

torch.cat 在 32 个张量的列表中
好的，谢谢。我在做torch.concatenate，但那是一件很麻烦的事情。

标签： python numpy opencv pytorch onnx

【解决方案1】：

您很可能使用 jit 模型，并且批量大小必须与模型训练时的大小完全相同。

t = torch.rand(1, 3, 256, 224)
t.size() # torch.Size([1, 3, 256, 224])
t2= t.expand(32, -1,-1,-1)
t2.size() # torch.Size([32, 3, 256, 224])

扩展张量不会分配新的内存，而只会在现有张量上创建一个新视图，然后您就可以得到所需的东西。只有张量步幅发生了变化。

【讨论】：