TensorFlow – How to create a tensor with all elements set to one
Last Updated :
30 Jan, 2023
TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
Methods Used:
- tf.ones: This methods accepts the shape and type and returns a tensor of given shape and type having all values set to 1.
- tf.fill: This method accepts shape, value and type and returns a tensor of given shape and type having all values set to value.
Example 1: This example uses ones() method to create a tensor with all elements set to one.
Python3
import tensorflow as tf
res = tf.ones(shape = ( 2 , 3 ))
print ( "Res: " , res )
|
Output:
Res: tf.Tensor(
[[1. 1. 1.]
[1. 1. 1.]], shape=(2, 3), dtype=float32)
Example 2: This example uses fill() method with value = 1 to create a tensor with all elements set to one.
Python3
import tensorflow as tf
res = tf.fill(dims = ( 2 , 3 ), value = 1 )
print ( "Res: " , res )
|
Output:
Res: tf.Tensor(
[[1 1 1]
[1 1 1]], shape=(2, 3), dtype=int32)
Share your thoughts in the comments
Please Login to comment...