Hi all. Could anyone tell me how to change the diagonal value to 1 for a deterministic matrix (pm.Deterministic)?

Basically I wish to implement formula like this:

The alpha matrix I constructed:

```
alpha_matrix = pm.Deterministic("alpha_matrix", C * pm.math.exp( (-1/2) * ((delta_t_matrix - mu_)/sigma_)**2 ) )
```

If I==j then I should set the value to 1. How to do that in pymc?