Find the trace of a square matrix.