如何在犰狳中创建nxn单位矩阵?

我正在尝试将犰狳用于C++。我使用该网站作为参考: Armadillo Link

arma::mat UnityMatrix = arma:mat(5,5,fill::zeros)
for (int ii = 0; ii < UnityMatrix.n_rows;ii++){
 for (int jj = 0; jj < UnityMatrix.n_cols;jj++){
   if (ii==jj){
     UnityMatrix(ii,jj)=1;
   }
 }

} 我创建一个由零组成的矩阵,然后在对角线上加一。这是可行的,但我相信在犰狳身上有更有效的方法。

我试过:

arma::umat InitM;
InitM.set_size(5,5);

这给了我一个5x5的随机值矩阵。