View source on GitHub
|
Creates a tensor with all elements set to one (1).
tf.ones(
shape,
dtype=tf.dtypes.float32,
name=None
)
See also tf.ones_like, tf.zeros, tf.fill, tf.eye.
This operation returns a tensor of type dtype with shape shape and
all elements set to one.
tf.ones([3, 4], tf.int32)<tf.Tensor: shape=(3, 4), dtype=int32, numpy=array([[1, 1, 1, 1],[1, 1, 1, 1],[1, 1, 1, 1]], dtype=int32)>
Args | |
|---|---|
shape
|
A list of integers, a tuple of integers, or
a 1-D Tensor of type int32.
|
dtype
|
Optional DType of an element in the resulting Tensor. Default is
tf.float32.
|
name
|
Optional string. A name for the operation. |
Returns | |
|---|---|
A Tensor with all elements set to one (1).
|