Python tf.RaggedTensor.nested_value_rowids用法及代碼示例

用法

nested_value_rowids(
    name=None
)

返回包含所有參差不齊維度的value_rowids 的元組。

rt.nested_value_rowids 是一個元組，其中包含 rt 中所有參差不齊的維度的 value_rowids 張量，從最外層到最內層排序。特別是 rt.nested_value_rowids = (rt.value_rowids(),) + value_ids 其中：

• value_ids = () 如果 rt.values 是 Tensor 。
• value_ids = rt.values.nested_value_rowids 否則。

例子：

rt = tf.ragged.constant(
    [[[[3, 1, 4, 1], [], [5, 9, 2]], [], [[6], []]]])
for i, ids in enumerate(rt.nested_value_rowids()):
  print('row ids for dimension %d:%s' % (i+1, ids.numpy()))
row ids for dimension 1:[0 0 0]
row ids for dimension 2:[0 0 0 2 2]
row ids for dimension 3:[0 0 0 0 2 2 2 3]