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