### 3x3 matrix
diagonal(x = 1, dim = c(3,3))
### 2x2x2 array
diagonal(x = 1:2, dim = c(2,2,2))
### extract diagonals
x <- diagonal(1:5, dim = c(5,5,5))
diagonal(x)
### set diagonals
x <- array(0, dim = c(2,2,2))
diagonal(x) <- 1:2
x
Run the code above in your browser using DataLab