Method square

tensorflow/contrib/eager/python/tfe_test.py:53–54

(x)

  def testGradients(self):

    def square(x):
      return math_ops.multiply(x, x)

    grad = tfe.gradients_function(square)
    self.assertEquals([6], [x.numpy() for x in grad(3.)])