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