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


Python tf.ragged.constant用法及代碼示例


從嵌套的 Python 列表構造一個常量 RaggedTensor。

用法

tf.ragged.constant(
    pylist, dtype=None, ragged_rank=None, inner_shape=None, name=None,
    row_splits_dtype=tf.dtypes.int64
)

參數

  • pylist 嵌套的 list , tuplenp.ndarray 。任何不是 list , tuplenp.ndarray 的嵌套元素必須是與 dtype 兼容的標量值。
  • dtype 返回的 RaggedTensor 的元素類型。如果未指定,則根據 pylist 中的標量值選擇默認值。
  • ragged_rank 一個整數,指定返回的 RaggedTensor 的不規則排名。必須為非負數且小於 K 。如果未指定 inner_shape,則默認為 max(0, K - 1)。如果指定了inner_shape,則默認為max(0, K - 1 - len(inner_shape))
  • inner_shape 一個整數元組,指定返回的 RaggedTensor 中各個內部值的形狀。如果未指定 ragged_rank,則默認為 ()。如果指定了 ragged_rank,則根據 pylist 的內容選擇默認值。
  • name 返回張量的名稱前綴(可選)。
  • row_splits_dtype 構造的 RaggedTensor 的 row_splits 的數據類型。 tf.int32tf.int64 之一。

返回

  • 一個可能參差不齊的張量,秩為 K 和指定的 ragged_rank ,包含來自 pylist 的值。

拋出

  • ValueError 如果pylist中的標量值嵌套深度不一致;或者如果 ragged_rank 或 inner_shape 與 pylist 不兼容。

例子:

tf.ragged.constant([[1, 2], [3], [4, 5, 6]])
<tf.RaggedTensor [[1, 2], [3], [4, 5, 6]]>

pylist 中的所有標量值必須具有相同的嵌套深度 K ,並且返回的 RaggedTensor 將具有排名 K 。如果 pylist 不包含標量值,則 Kpylist 中空列表的最大深度大一。 pylist 中的所有標量值必須與 dtype 兼容。

相關用法


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