标引火炬张量

Question

我有一个Pytorch代码，它在for循环的每次迭代中生成一个Pytorch张量，它们的大小都相同。我想将每个张量分配给新张量的行，新张量将在末尾包含所有张量。在其他作品中，像这样

for i=1:N:
  X = torch.Tensor([[1,2,3], [3,2,5]])
  #Y is a pytorch tensor
  Y[i] = X

我想知道如何用Pytorch实现它。

Answer 1

您可以使用torch.cat连接张量：

tensors = []
for i in range(N):
  X = torch.tensor([[1,2,3], [3,2,5]])
  tensors.append(X)
Y = torch.cat(tensors, dim=0)  # dim 0 is the rows of the tensor