Tensorflow.js tf.onesLike() Function

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

Reference: https://js.tensorflow.org/api/latest/#onesLike

My Personal Notes arrow_drop_up

Save