A transpose version also for the CPU.
Updating the test.