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

生成与分段对应的 RaggedTensor row_splits。

用法

tf.ragged.segment_ids_to_row_splits(
    segment_ids, num_segments=None, out_type=None, name=None
)

segment_ids 一维整数张量。
num_segments 一个标量整数，指示段数。默认为 max(segment_ids) + 1(如果 segment_ids 为空，则为零)。
out_type 返回值的 dtype。默认为 segment_ids.dtype 或 tf.int64 如果 segment_ids 没有数据类型。
name 返回张量的名称前缀(可选)。

返回一个整数向量 splits ，其中 splits[0] = 0 和 splits[i] = splits[i-1] + count(segment_ids==i) 。例子：

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

相关用法

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