How do I create a program that reads a matrix array and returns me a second matrix composed of zeros and the main diagonal of the first matrix?

Example:

Answer 1

[Guillaume]

b = diag(diag(a));

[Rômulo Prazeres]

Thanks Guillaume!