迭代追加1D张量以制作2D张量

Question

我有这个长度为64的张量，我试图附加到第一个维度上以递归方式制作2D张量。本质上，每次迭代都会创建一个新的1D 64长度张量，并且迭代6次，因此最后我应该拥有一个6 x 64张量。我该怎么做？

我尝试过

names_tensor = torch.cat((names_tensor, sampled_indexes), dim=1)

其中name_tensor以torch.zeros(0)发起，而sampled_indexes是每次迭代都附加的64长度张量

Answer 1

import torch
names_tensor = torch.zeros(0)
sampled_indexes = torch.ones(64)

names_tensor = torch.cat((names_tensor, sampled_indexes.repeat(6))).reshape(6,64)
names_tensor.size()
>> torch.Size([6, 64])

首先使用torch.cat()来连接张量序列，然后使用torch.reshape()来重整张量