[gru_diag]