Python tf.ones_like用法及代碼示例

創建一個與輸入具有相同形狀的所有張量。

用法

tf.ones_like(
    input, dtype=None, name=None
)

參數

input 一個Tensor。
dtype 返回的 Tensor 的類型。必須是 float16 , float32 , float64 , int8 , uint8 , int16 , uint16 , int32 , int64 , complex64 , complex128 , bool 或 string 。
name 操作的名稱(可選)。

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

另見tf.ones。

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

例如：

tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
tf.ones_like(tensor)
<tf.Tensor:shape=(2, 3), dtype=int32, numpy=
  array([[1, 1, 1],
         [1, 1, 1]], dtype=int32)>

相關用法

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