在张量流中取消参差不齐的张量
Unnest ragged tensor in tensorflow
我关注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
.
的组合并不走运
有人知道答案吗?
经过一番搜索,原来有一个内置方法:
rt.merge_dims(1,2)
.
我关注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
.
有人知道答案吗?
经过一番搜索,原来有一个内置方法:
rt.merge_dims(1,2)
.