x <- array(1, dim = c(3,3,3))
diag3Darray(x) <- c(2,2,2)
x
# , , 1
#
# [,1] [,2] [,3]
# [1,] 2 1 1
# [2,] 1 1 1
# [3,] 1 1 1
#
# , , 2
#
# [,1] [,2] [,3]
# [1,] 1 1 1
# [2,] 1 2 1
# [3,] 1 1 1
#
# , , 3
#
# [,1] [,2] [,3]
# [1,] 1 1 1
# [2,] 1 1 1
# [3,] 1 1 2
Run the code above in your browser using DataLab