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
import tensorflow as tf
input = tf.constant([[ 1 , 2 , 3 ], [ 4 , 5 , 6 ]])
res = tf.ones_like( input )
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
import tensorflow as tf
input = tf.constant([[ 1 , 2 , 3 ], [ 4 , 5 , 6 ]])
print ( "Input: " , input )
res = tf.ones_like( input , dtype = tf.float64)
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)
Last Updated :
01 Aug, 2020
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...