use corrected shape striding
Merge pull request #4 from thouis/fix_striding

use corrected shape striding