training_ops_test.py 文件源码

def testApplyAdam(self):
    for dtype, force_gpu in itertools.product(
        [np.complex64], [True]):
      var = np.arange(100).astype(dtype)
      m = np.arange(1, 101).astype(dtype)
      v = np.arange(101, 201).astype(dtype)
      grad = np.arange(100).astype(dtype)
      self._testTypesForAdam(var, m, v, grad, force_gpu)