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

Python tf.RaggedTensor.row_starts用法及代碼示例


用法

row_starts(
    name=None
)

參數

  • name 返回張量的名稱前綴(可選)。

返回

  • 形狀為 [nrows] 的一維整數張量。返回的張量是非負的，並按升序排序。

返回此不規則張量中行的起始索引。

這些索引指定每行的值從 self.values 開始的位置。 rt.row_starts() 等於 rt.row_splits[:-1] 。

例子：

rt = tf.ragged.constant([[3, 1, 4, 1], [], [5, 9, 2], [6], []])
print(rt.values)
tf.Tensor([3 1 4 1 5 9 2 6], shape=(8,), dtype=int32)
print(rt.row_starts())  # indices of row starts in rt.values
tf.Tensor([0 4 4 7 8], shape=(5,), dtype=int64)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.RaggedTensor.row_starts。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。