TensorFlow 1 version | View source on GitHub |
Creates a tensor with all elements set to 1.
tf.ones(
shape, dtype=tf.dtypes.float32, name=None
)
This operation returns a tensor of type dtype
with shape shape
and all
elements set to 1.
For example:
tf.ones([2, 3], tf.int32) # [[1, 1, 1], [1, 1, 1]]
Args | |
---|---|
shape
|
A list of integers, a tuple of integers, or a 1-D Tensor of type
int32 .
|
dtype
|
The type of an element in the resulting Tensor .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor with all elements set to 1.
|