tf.ones
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). |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/ones