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]