Skip to content
Related Articles
TensorFlow – How to create a tensor of all ones that has the same shape as the input tensor
• Last Updated : 01 Aug, 2020

TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning  neural networks.

Method Used:

• ones_like: This method accepts a Tensor as input and returns a Tensor with same shape having all values set to one.

Example 1:

## Python3

 `# importing the library``import` `tensorflow as tf`` ` `# Initializing the Input``input` `=` `tf.constant([[``1``, ``2``, ``3``], [``4``, ``5``, ``6``]])`` ` `# Generating Tensor with having all values as 1``res ``=` `tf.ones_like(``input``)`` ` `# Printing the resulting Tensors``print``(``"Res: "``, res )`

Output:

```Res:  tf.Tensor(
[[1 1 1]
[1 1 1]], shape=(2, 3), dtype=int32)

```

Example 2: This example explicitly specifies the type of the resulting tensor.

## Python3

 `# importing the library``import` `tensorflow as tf`` ` `# Initializing the Input``input` `=` `tf.constant([[``1``, ``2``, ``3``], [``4``, ``5``, ``6``]])`` ` `# Printing the Input``print``(``"Input: "``, ``input``)`` ` `# Generating Tensor with having all values as 1``res ``=` `tf.ones_like(``input``, dtype ``=` `tf.float64)`` ` `# Printing the resulting Tensors``print``(``"Res: "``, res )`

Output:

```Input:  tf.Tensor(
[[1 2 3]
[4 5 6]], shape=(2, 3), dtype=int32)
Res:  tf.Tensor(
[[1. 1. 1.]
[1. 1. 1.]], shape=(2, 3), dtype=float64)

```

Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course

My Personal Notes arrow_drop_up