在张量流中取消参差不齐的张量

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).

在张量流中取消参差不齐的张量

Unnest ragged tensor in tensorflow

python

tensorflow

ragged