Tensorflow.js tf.onesLike() Function
Last Updated :
10 May, 2021
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
import * as tf from "@tensorflow/tfjs"
const x = tf.tensor([1, 2]);
tf.onesLike(x).print();
|
Output:
Tensor
[1, 1]
Example 2:
Javascript
import * as tf from "@tensorflow/tfjs"
const x = tf.tensor2d([8, 2, 5, 6], [2, 2]);
tf.onesLike(x).print();
|
Output:
Tensor
[[1, 1],
[1, 1]]
Reference: https://js.tensorflow.org/api/latest/#onesLike
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...