当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.ragged.row_splits_to_segment_ids用法及代码示例


生成对应于 RaggedTensor row_splits 的分段。

用法

tf.ragged.row_splits_to_segment_ids(
    splits, name=None, out_type=None
)

参数

  • splits 排序的一维整数张量。 splits[0] 必须为零。
  • name 返回张量的名称前缀(可选)。
  • out_type 返回值的 dtype。默认为 splits.dtype 或 tf.int64 如果 splits 没有数据类型。

返回

  • 排序后的一维整数张量，带有shape=[splits[-1]]

抛出

  • ValueError 如果splits 无效。

返回一个整数向量 segment_ids ，其中 segment_ids[i] == j 如果 splits[j] <= i < splits[j+1] 。例子：

print(tf.ragged.row_splits_to_segment_ids([0, 3, 3, 5, 6, 9]))
 tf.Tensor([0 0 0 2 2 3 4 4 4], shape=(9,), dtype=int64)

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.ragged.row_splits_to_segment_ids。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。