The "shake to undo" feature is actually very old, I remember discovering it around 2010. It works in all text fields system-wide, not just in the Notes app!