Tensorflow.js is an open-source library developed by Google for running machine learning models and deep learning neural networks in the browser or node environment.
The tf.onesLike() function is used to create a tf.Tensor with all elements set to 1 with the same shape as the given tensor.
Syntax:
tf.onesLike (x)
Parameters:
- x: a tf.Tensor whose all elements will be set as 1.
Return Value: It returns a tf.Tensor.
Example 1:
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Creating the tensor
const x = tf.tensor([1, 2]);
// Printing the tensor after setting the value of tensor to 1
tf.onesLike(x).print();
Output:
Tensor [1, 1]
Example 2:
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Creating the tensor
const x = tf.tensor2d([8, 2, 5, 6], [2, 2]);
// Printing the tensor after setting the value of tensor to 1
tf.onesLike(x).print();