pytorch 三角行列を作れない

Question

x=[1,1,1,]
y=[2,2,2,]

この状態から

a=[[2,0,0,],
   [1,2,0,],
   [1,1,2,]]

こうしたいです

torch.trilとtorch.diagというのを見つけましたがうまくいかず困ってます
やり方を教えてください

Answer 1

しつもんしなおします