Skip to content
Related Articles

Related Articles

TensorFlow – How to create a tensor of all ones that has the same shape as the input tensor

View Discussion
Improve Article
Save Article
  • 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)



My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!