[draft] add optional return_2d flag for including 2D results in predict() return

[draft] add optional return_2d flag for including 2D results in predict() return #41