tf.ones

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

View aliases

Compat aliases for migration

See Migration guide for more details.

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

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