compressed_matrix: Improved handling of dimensions for STL matrix.

When copying back from device, there's no need for implicitly assuming a square matrix,
because the dimensions are known on the device.