Implementation nlinalg.alloc_diag() in Aesara tensor

Thanks! Now, I can compile the model.