Skip to content
Related Articles

Related Articles

TensorFlow – How to create a tensor with all elements set to one
  • Last Updated : 01 Aug, 2020

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 give 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




# importing the library
import tensorflow as tf
  
# Generating a Tensor of shape (2, 3)
res = tf.ones(shape = (2, 3))
  
# Printing the resulting Tensors
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




# importing the library
import tensorflow as tf
  
# Generating a Tensor of shape (2, 3)
res = tf.fill(dims = (2, 3), value = 1)
  
# Printing the resulting Tensors
print("Res: ", res )

Output:


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

 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
Recommended Articles
Page :