TensorFlow函數(shù)：tf.ones

tf.ones 函數(shù)

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

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

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

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

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

例如：

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

參數(shù)：

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

返回值：

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