Skip to content
Related Articles

Related Articles

TensorFlow – How to create a tensor with all elements set to one

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.

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)

My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!