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

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


用法

row_lengths(
    axis=1, name=None
)

参数

  • axis 一个整数常量，指示应返回其行长的轴。
  • name 返回张量的名称前缀(可选)。

返回

  • 形状可能参差不齐的整数张量 self.shape[:axis] 。

抛出

  • ValueError 如果axis 超出范围。

返回此不规则张量中行的长度。

rt.row_lengths()[i] 表示 rt 的第 i 行中的值的数量。

例子：

rt = tf.ragged.constant(
    [[[3, 1, 4], [1]], [], [[5, 9], [2]], [[6]], []])
print(rt.row_lengths())  # lengths of rows in rt
tf.Tensor([2 0 2 1 0], shape=(5,), dtype=int64)
print(rt.row_lengths(axis=2))  # lengths of axis=2 rows.
<tf.RaggedTensor [[3, 1], [], [2, 1], [1], []]>

相关用法


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