Not sure yet, but do we need this unmap transpose? Doesn't it work with just the reshape?
Having a 4D cube test with 2 input coords would probably help confirm this...