TensorFlow函數：tf.ones

tf.ones 函數

ones(
    shape,
    dtype=tf.float32,
    name=None
)

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

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

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

這個操作返回一個 dtype 類型的張量,並且形狀為 shape,所有的元素都被設置為1.

例如：

tf.ones([2, 3], tf.int32) ==> [[1, 1, 1], [1, 1, 1]]

參數：

• shape：整數的列表,整數的元組或 int32 類型的一維張量.
• dtype：生成的張量中元素的類型.
• name：操作的名稱(可選).

返回值：

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