Como posso atribuir um valor às diagonais de uma matriz 4-D usando indexação linear no MATLA

Tenho uma matriz 4-DA do tamanhoNxNxPxQ. Como posso alterar facilmente os valores da diagonal para 1 para cadaNxN Submatriz 2-D de maneira vetorizad