Implementation nlinalg.alloc_diag() in Aesara tensor

Aesara also has at.fill_diagonal() which may help.

2 Likes