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:
Javascript
// 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:
Javascript
// 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(); |
Output:
Tensor
[[1, 1],
[1, 1]]