TensorFlow函數(shù)：tf.ones_like

tf.ones_like 函數(shù)

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

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

請(qǐng)參閱指南：生成常量,序列和隨機(jī)值>常量值張量

創(chuàng)建一個(gè)將所有元素都設(shè)置為1的張量.

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

例如：

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

參數(shù)：

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

返回值：

該函數(shù)將返回所有元素設(shè)置為1的張量.