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

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

用法

value_rowids(
    name=None
)

參數

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

返回

  • 形狀為 self.values.shape[:1] 的一維整數 Tensor 。返回的張量是非負的，並按升序排序。

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

rt.value_rowids() 對應 one-to-one 與 rt.values 的最外層維度，並指定包含每個值的行。特別是，rt[row] 行由值 rt.values[j] 組成，其中 rt.value_rowids()[j] == row 。

例子：

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.value_rowids())  # corresponds 1:1 with rt.values
tf.Tensor([0 0 0 0 2 2 2 3], shape=(8,), dtype=int64)

相關用法


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