Tensorflow.js tf.round() Function
Last Updated :
10 May, 2021
Tensorflow.js is an open-source library that is being developed by Google for running machine learning models as well as deep learning neural networks in the browser or node environment.
The .round() function is used to find the round value of the stated tensor input, and it is done element wise. Moreover, it is helpful in implementing banker’s rounding.
Syntax :
tf.round(x)
Parameters: This function accepts a parameter which is illustrated below:
- x: It is the tensor input whose round value is to be computed and it can be of type tf.Tensor, or TypedArray, or Array.
Return Value: It returns the tf.Tensor object.
Example 1: In this example, we are defining an input tensor of float type and then printing the round value of it. For creating an input tensor we are utilizing the .tensor1d() method and in order to print the output we are using the .print() method.
Javascript
import * as tf from "@tensorflow/tfjs"
const y = tf.tensor1d([5.3, 1.4, 4.5, 9.4]);
y.round().print();
|
Output:
Tensor
[5, 1, 4, 9]
Example 2: In this example, double type values are considered as tensor input and the parameter is passed directly to the round function.
Javascript
var val = [8.69967679, 0.999999, 78.7823826382];
const y = tf.tensor1d(val);
var res = tf.round(y)
res.print();
|
Output:
Tensor
[9, 1, 79]
Share your thoughts in the comments
Please Login to comment...