fixed_discrete_covariance: get diagonal indices faster

Profiling repeated calls to `group_model.model()` (to get learning curves), _way_ too much time was being spent on `diagonal = diag(K)`. This way doesn't copy the diagonal.