variable_ops_test.py 文件源码

def testTemporaryVariable(self):
    with self.test_session(use_gpu=True):
      var = gen_state_ops._temporary_variable(
          [1, 2],
          tf.complex64,
          var_name="foo")
      var = tf.assign(var, [[4.0+5.0j, 5.0+4.0j]])
      var = tf.assign_add(var, [[6.0+7.0j, 7.0+6.0j]])
      final = gen_state_ops._destroy_temporary_variable(var, var_name="foo")
      self.assertAllClose([[10.0+12.0j, 12.0+10.0j]], final.eval())