示例#1
0
 def test_is_scalar_with_scalar_tf_variable(self):
     self.assertTrue(tensor_utils.is_scalar(tf.Variable(0.0, 'scalar')))
示例#2
0
 def test_is_scalar_with_nonscalar_tf_variable(self):
     self.assertFalse(
         tensor_utils.is_scalar(tf.Variable([0.0, 1.0], 'notscalar')))
示例#3
0
 def test_is_scalar_with_tf_constant(self):
     self.assertTrue(tensor_utils.is_scalar(tf.constant(10)))