张量流中未嵌套的参差不齐的张量答案

【问题标题】：Unnest ragged tensor in tensorflow张量流中未嵌套的参差不齐的张量
【发布时间】：2021-02-23 09:36:05
【问题描述】：

我关注tf.RaggedTensorrt：

<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 的组合我没有运气。

有人知道答案吗？

【问题讨论】：

标签： python tensorflow ragged

【解决方案1】：

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

【讨论】：