restriction to the diagonal