tf.ones

Creates a tensor with all elements set to one (1).

View aliases

Compat aliases for migration

tf.compat.v1.ones

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

Used in the notebooks

Used in the guide	Used in the tutorials
Validating correctness & numerical equivalence Use TF1.x models in TF2 workflows Better performance with tf.function Extension types Ragged tensors	Multilevel Modeling Primer in TensorFlow Probability Learnable Distributions Zoo Probabilistic PCA Policies Bayesian Switchpoint Analysis

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).

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/ones