For vector inputs, `pullback_function` currently returns a pullback that returns `n x 1` matrices instead of vectors, when it should return vectors.