Python is_variable Examples

Example #1

File: gradient.py Project: xiaoguozhi/probability

 def c2t(x):
     # Don't use convert_nonref_to_tensor here. We want to have semantics like
     # tf.GradientTape which watches only trainable_variables. (Note: we also
     # don't want to cal c2t on non-trainable variables since these are already
     # watchable by GradientTape.)
     if tensor_util.is_module(x) or tensor_util.is_variable(x):
         return x
     i[0] += 1
     return tf.convert_to_tensor(x,
                                 dtype_hint=tf.float32,
                                 name='x{}'.format(i[0]))

Example #2

File: tensor_util_test.py Project: vishalbelsare/probability

 def test_is_variable(self):
   self.assertTrue(tensor_util.is_variable(tf.Variable(0.)))
   self.assertTrue(tensor_util.is_variable(tf.Variable(0., trainable=False)))