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


Python tf.RaggedTensor.row_starts用法及代码示例


用法

row_starts(
    name=None
)

参数

  • name 返回张量的名称前缀(可选)。

返回

  • 形状为 [nrows] 的一维整数张量。返回的张量是非负的,并按升序排序。

返回此不规则张量中行的起始索引。

这些索引指定每行的值从 self.values 开始的位置。 rt.row_starts() 等于 rt.row_splits[:-1]

例子:

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.row_starts())  # indices of row starts in rt.values
tf.Tensor([0 4 4 7 8], shape=(5,), dtype=int64)

相关用法


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