The result should be [2., 6.,]. I clarify it in the article. Thanks. But x has the shape (2,) so dy/dx has the shape of (2,) too. You mistake x to have one element only.