python - 张量流中未嵌套的参差不齐的张量

Question

我有以下内容tf.RaggedTensor rt：

<tf.RaggedTensor [[[7592]], [[7592], [2088]], [[7592], [2045], [2236]], [[7592], [2045], [2236], [6358, 16429, 2072]]]>

这是密集版本：

array([[[ 7592,     0,     0],
        [    0,     0,     0],
        [    0,     0,     0],
        [    0,     0,     0]],

       [[ 7592,     0,     0],
        [ 2088,     0,     0],
        [    0,     0,     0],
        [    0,     0,     0]],

       [[ 7592,     0,     0],
        [ 2045,     0,     0],
        [ 2236,     0,     0],
        [    0,     0,     0]],

       [[ 7592,     0,     0],
        [ 2045,     0,     0],
        [ 2236,     0,     0],
        [ 6358, 16429,  2072]]], dtype=int32)>

我需要“ unnest”rt对此：

<tf.RaggedTensor [[7592], [7592,2088], [7592,2045,2236], [7592,2045,2236,6358, 16429, 2072]]>

但我对tf.concat,tf.ragged.map_flat_values或的组合没有运气tf.squeeze。

有人知道答案吗？

Answer 1

经过一番搜索，原来有一个内置的方法： rt.merge_dims(1,2).