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

Python tf.ones用法及代碼示例


創建一個所有元素都設置為一 (1) 的張量。

用法

tf.ones(
    shape, dtype=tf.dtypes.float32, name=None
)

參數

  • shape 整數的 list、整數的 tuple 或 int32 類型的一維 Tensor 。
  • dtype 生成的 Tensor 中元素的可選 DType。默認為 tf.float32 。
  • name 可選字符串。操作的名稱。

返回

  • 一個Tensor，所有元素都設置為一 (1)。

另見tf.ones_like、tf.zeros、tf.fill、tf.eye。

此操作返回一個類型為 dtype 的張量，形狀為 shape，所有元素都設置為 1。

tf.ones([3, 4], tf.int32)
<tf.Tensor:shape=(3, 4), dtype=int32, numpy=
array([[1, 1, 1, 1],
       [1, 1, 1, 1],
       [1, 1, 1, 1]], dtype=int32)>

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.ones。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。