tf.string_split表示字符串张量

Question

我有一个看起来像

的张量

x = [b'1 2 3 4 5'
     b'3 2 1'
     b'1 2']

如何将此张量的单个字符串拆分为数字？当我应用tf.string_split(x)时，我收到以下错误：

TypeError: Failed to convert object of type <class 'tensorflow.python.framework.sparse_tensor.SparseTensor'> to Tensor. Contents: SparseTensor(indices=Tensor("StringSplit_1:0", shape=(?, 2), dtype=int64), values=Tensor("StringSplit_1:1", shape=(?,), dtype=string), dense_shape=Tensor("StringSplit_1:2", shape=(2,), dtype=int64)). Consider casting elements to a supported type.

当我用tf.string_split(x).values尝试时，我得到以下张量： x = [b'1' b'2' b'3' b'4' b'5' b'3 b'2 b'1' b'1 b'2']

这不是我想要的，因为数字属于一起的信息已经消失。我想要的是这样的东西

x = [[b'1' b'2' b'3' b'4' b'5']
     [b'3' b'2' b'1']
     [b'1' b'2']]

在下一步中将tf.string_to_number转换为int32。

有人能帮助我吗？