當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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