You can’t manually cache purchases, but don’t worry—when you play an album while connected to Wi-Fi, all its tracks will be automatically cached for offline listening.
How can I check what’s cached?
Turn on Airplane Mode. Any albums or tracks that aren’t cached will appear greyed out in your collection.