TensorFlow函數：tf.ones_like

tf.ones_like 函數

ones_like(
    tensor,
    dtype=None,
    name=None,
    optimize=True
)

定義在：tensorflow/python/ops/array_ops.py.

請參閱指南：生成常量,序列和隨機值>常量值張量

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

給定一個張量(tensor),此操作返回與所有元素設置為1的張量相同的類型和形狀的張量.可選地,可以為返回的張量指定一個新的類型(dtype).

例如：

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

參數：

• tensor：張量.
• dtype：返回的 Tensor 類型.必須是：float32,float64,int8,int16,int32,int64,uint8,complex64,complex128或bool.
• name：操作的名稱(可選).
• optimize：如果為 true,則嘗試靜態確定 "張量" 的形狀並將其編碼為常量.

返回值：

該函數將返回所有元素設置為1的張量.