lambdify preserves arg shape