Python Pytorch如何增加批量

我现在有一个torch.Size的张量（[1,3,256,224]），但我需要它成为输入形状[32,3,256,224]。我正在实时捕获数据，所以dataloader似乎不是一个好的选择。有没有简单的方法可以将32个torch.size（[1,3,256,224]）合并为1个大小为[32,3,256,224]的张量？

您可能使用的是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])

---

扩展一个张量不会分配新的内存，只会在现有的张量上创建一个新的视图，您就可以得到所需的。只更改了张量步幅。

torch.cat

在32个张量的列表中是的，谢谢。我在做火炬连接，但那是件很重要的事。