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
# importing the libraryimport tensorflow as tf# Generating a Tensor of shape (2, 3)res = tf.ones(shape = (2, 3))# Printing the resulting Tensorsprint("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 libraryimport tensorflow as tf# Generating a Tensor of shape (2, 3)res = tf.fill(dims = (2, 3), value = 1)# Printing the resulting Tensorsprint("Res: ", res ) |
Output:
Res: tf.Tensor( [[1 1 1] [1 1 1]], shape=(2, 3), dtype=int32)
… [Trackback]
[…] Info to that Topic: geeksforgeeks.org/tensorflow-how-to-create-a-tensor-with-all-elements-set-to-one/ […]