Method f

tensorflow/python/eager/backprop_test.py:130–138

(x)

    @decorator
    def f(x):
      x1 = array_ops.identity(x)
      with backprop.GradientTape() as t:
        t.watch(x)
        t.watch(x1)
        y1 = x * 2.
        y2 = x1 * 3.
        loss = y1 + y2
      return t.gradient(loss, [x, x1])

    self.assertAllClose([2., 3.], f(constant_op.constant(10.)))