當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。