lua - Torch：连接不同维度的张量

Question

我有一个x_at_i = torch.Tensor(1,i)在 i = 0 到 n 的每次迭代中增长。我想将所有不同大小的张量连接成一个矩阵，并用零填充剩余的单元格。什么是最惯用的方法。例如：

x_at_1 = 1
x_at_2 = 1 2
x_at_3 = 1 2 3
x_at_4 = 1 2 3 4
X = torch.cat(x_at_1, x_at_2, x_at_3, x_at_4)
X = [ 1 0 0 0
      1 2 0 0
      1 2 3 0
      1 2 3 4 ]

Answer 1

如果您知道并假设您可以在每次迭代中轻松n访问您的内容，我会尝试类似x_at_i

X = torch.Tensor(n, n):zero()
for i = 1, n do
    X[i]:narrow(1, 1, i):copy(x_at[i])
end