Creates a tensor with all elements set to one (1).
tf.ones(
shape,
dtype=tf.dtypes.float32
,
name=None,
layout=None
)
Used in the notebooks
Used in the guide |
Used in the tutorials |
|
|
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.
|
layout
|
Optional, tf.experimental.dtensor.Layout . If provided, the result
is a DTensor with the
provided layout.
|
Returns |
A Tensor with all elements set to one (1).
|