当前位置: 首页>>编程示例 >>用法及示例精选 >>正文

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 , tuple 或 np.ndarray 。任何不是 list , tuple 或 np.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.int32 或 tf.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 不包含标量值，则 K 比 pylist 中空列表的最大深度大一。 pylist 中的所有标量值必须与 dtype 兼容。

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.ragged.constant。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。