Don't return `vec(y)` in `withjacobian` - #1568

mcabbott wants to merge 3 commits into

Don't return `vec(y)` in `withjacobian`#1568
mcabbott wants to merge 3 commits into
FluxML:masterfrom
mcabbott:withjacobian_shape

fix vec(y) in withjacobian