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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。