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

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


用法

row_lengths(
    axis=1, name=None
)

參數

  • axis 一個整數常量，指示應返回其行長的軸。
  • name 返回張量的名稱前綴(可選)。

返回

  • 形狀可能參差不齊的整數張量 self.shape[:axis] 。

拋出

  • ValueError 如果axis 超出範圍。

返回此不規則張量中行的長度。

rt.row_lengths()[i] 表示 rt 的第 i 行中的值的數量。

例子：

rt = tf.ragged.constant(
    [[[3, 1, 4], [1]], [], [[5, 9], [2]], [[6]], []])
print(rt.row_lengths())  # lengths of rows in rt
tf.Tensor([2 0 2 1 0], shape=(5,), dtype=int64)
print(rt.row_lengths(axis=2))  # lengths of axis=2 rows.
<tf.RaggedTensor [[3, 1], [], [2, 1], [1], []]>

相關用法


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