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


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。