result_types_test.py 文件源码

def test_tensor_shape(self):
    self.assertConverts(tf.TensorShape([]), tdt.TensorType(()))
    self.assertConverts(tf.TensorShape([1]), tdt.TensorType((1,)))
    self.assertConverts(tf.TensorShape([1, 2]), tdt.TensorType((1, 2)))
    self.assertConverts(tf.TensorShape([1, 2, 3]), tdt.TensorType((1, 2, 3)))