在tensorflow中Unnest褴褛张量

Question

我有以下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";rtto this:

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