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

用法

@classmethod
from_row_lengths(
    values, row_lengths, name=None, validate=True
)

參數

values 形狀可能參差不齊的張量 [nvals, ...] 。
row_lengths 形狀為 [nrows] 的一維整數張量。必須是非負數。 sum(row_lengths) 必須是 nvals 。
name RaggedTensor 的名稱前綴(可選)。
validate 如果為真，則使用斷言檢查參數是否形成有效的 RaggedTensor 。注意：這些斷言會產生運行時成本，因為必須檢查每個張量值。

一個RaggedTensor。 result.rank = values.rank + 1 。 result.ragged_rank = values.ragged_rank + 1 。

創建一個 RaggedTensor ，其中行由 row_lengths 分區。

返回的 RaggedTensor 對應於由以下定義的 python 列表：

result = [[values.pop(0) for i in range(length)]
          for length in row_lengths]

例子：

print(tf.RaggedTensor.from_row_lengths(
    values=[3, 1, 4, 1, 5, 9, 2, 6],
    row_lengths=[4, 0, 3, 1, 0]))
<tf.RaggedTensor [[3, 1, 4, 1], [], [5, 9, 2], [6], []]>

相關用法

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