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


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


用法

nrows(
    out_type=None, name=None
)

参数

  • out_type dtype 用于返回的张量。默认为 self.row_splits.dtype
  • name 返回张量的名称前缀(可选)。

返回

  • 具有 dtype out_type 的标量 Tensor

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

即张量最外层维度的大小。

例子:

rt = tf.ragged.constant([[3, 1, 4, 1], [], [5, 9, 2], [6], []])
print(rt.nrows())  # rt has 5 rows.
tf.Tensor(5, shape=(), dtype=int64)

相关用法


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