用matlab编写函数式文件,计算任意方阵的主对角元素的和,要求对输入矩阵进行判断

A=input('请输入您的矩阵,例如输入 [1:3;2:4] ：');
[m,n]=size(A);
while =n
disp(':');
A=input('您输入的矩阵不是方阵,请重新输入一个方阵：')
[m,n]=size(A);
end
sum=trace(A)