lambdify preserves arg shape

fixes #2790