[BugFix] Avoid reshape(-1) for inputs to objectives modules (#2494)

[BugFix] Avoid reshape(-1) for inputs to objectives modules (#2494) #3088