Torche: Ténors de concaténation de différentes dimensions

Question

J'ai un x_at_i = torch.Tensor(1,i) qui augmente à chaque itération où i = 0 à n. Je voudrais concaténer tous les tenseurs de différentes tailles dans une matrice et remplir les cellules restantes avec des zéros. Quelle est la manière la plus idiomatique de cela? Par exemple:

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

Si vous connaissez n et en supposant que vous avez accès à votre x_at_i facilement à chaque itération je voudrais essayer quelque chose comme

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