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

Python tf.zeros用法及代碼示例


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

用法

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

參數

  • shape 整數的 list、整數的 tuple 或 int32 類型的一維 Tensor 。
  • dtype 結果 Tensor 中元素的 DType。
  • name 可選字符串。操作的名稱。

返回

  • 一個Tensor，所有元素都設置為零。

另見tf.zeros_like、tf.ones、tf.fill、tf.eye。

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

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

相關用法


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