Returns the unit matrix or the specified dimension