TensorFlow – How to create a tensor of all ones that has the same shape as the input tensor
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)
