Есть ли способ использоватьdiag() функция в матрице без использования встроенной функции или итерации?

   M<-matrix(1:9, ncol=3) # make a matrix 

    q5b<-function(M){ #function

    }

я знаю этоM[1,1], M[2,2], а такжеM[3,3] даст мне такой же вывод, какdiag(M), Тем не менее, я не могу придумать способ сделать это без цикла for.

Мой мыслительный процесс был у меня должно быть состояние, при которомrow index == column index затем в матрице выведите это значение. Я ценю любые предложения.