Method test_with_custom_gradient

test/unit/test_gradient.py:21–25

(self)

    self.assertListEqual(x.sum().gradient(w)[0].tolist(), [0.0, 0.0, 0.0])

  def test_with_custom_gradient(self):
    x = Tensor([1.0, 2.0, 3.0])
    z = (x * x).sum()
    dx = z.gradient(x, gradient=Tensor([3.0]))[0]
    self.assertListEqual(dx.tolist(), [6.0, 12.0, 18.0])

  def test_broadcast_gradient(self):
    x = Tensor([[1.0], [2.0], [3.0]])