3 ms·

Yeah, you got it. The diagonal() matrix is not actually created since you really don't need all the elements in the final computation.