The original line will give a wrong result. Maybe it's due to numpy's api is changed