我有一个像A这样的矩阵,我想计算这个矩阵的每个对角线的总和,并在像Y这样的向量中显示它.
A=[1 2 3; 4 5 6; 7 8 9]
Y=[3 8 15 12 7]
我知道代码
[sum(diag(y,2)),sum(diag(y,1)),sum(diag(y,0)),sum(diag (y,-1)),sum(diag (y,-2))]
但我想把它写成一个函数.