Maybe this should be a new dataset class? PatchesWithPosition(DenseDesignMatrix)?