python - 从张量列表创建参差不齐的张量

Question

我想从 TensorFlow 2.0 中的张量列表创建一个参差不齐的张量，如下所示:

a = tf.convert_to_tensor([1,2])
b = tf.convert_to_tensor([1,2,3])
tf.ragged.constant([a, b])

但这会抛出 ValueError: TypeError: Scalar tensor has no `len()`。另一方面，以下代码从列表的列表中创建参差不齐的张量，但运行良好。

a = [1,2]
b = [1,2,3]
tf.ragged.constant([a,b])

有没有什么方法可以直接从张量列表创建参差不齐的张量，而无需先将张量转换为 Python 列表？

Answer 1

你可以使用tf.ragged.stack:

>>> a = tf.convert_to_tensor([1,2])
>>> b = tf.convert_to_tensor([1,2,3])
>>> tf.ragged.stack([a, b])
<tf.RaggedTensor [[1, 2], [1, 2, 3]]>