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

Python tf.compat.v1.ones_like用法及代碼示例

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

用法

tf.compat.v1.ones_like(
    tensor, dtype=None, name=None, optimize=True
)

參數

  • tensor 一個Tensor。
  • dtype 返回的 Tensor 的類型。必須是 float32 , float64 , int8 , uint8 , int16 , uint16 , int32 , int64 , complex64 , complex128 或 bool 。
  • name 操作的名稱(可選)。
  • optimize 如果為真，嘗試靜態確定'tensor' 的形狀並將其編碼為常數。

返回

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

另見tf.ones。

給定單個張量 (tensor)，此操作返回與 tensor 具有相同類型和形狀的張量，所有元素都設置為 1。您可以選擇為返回的張量指定新類型 (dtype)。

例如：

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor)  # [[1, 1, 1], [1, 1, 1]]

相關用法


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