Trying to do reconstruction with `inverse_transform`, instead of doing
it manually. Having some confusion with 3D array indexing, though, so
the output is garbage.